bugfix on load_parquet()

earthview.py

@@ -60,10 +60,11 @@ def load_dataset(subset, dataset="satellogic/EarthView", split="train", shards =
 
 def load_parquet(subset_or_filename, batch_size=100):
     if subset_or_filename in get_subsets():
-        pqfile = ParquetFile(f"dataset/{subset_or_filename}/sample.parquet")
+        filename = f"dataset/{subset_or_filename}/sample.parquet"
     else:
-        pqfile = subset_or_filename
+        filename = subset_or_filename
 
+    pqfile = ParquetFile(filename)
     batch  = pqfile.iter_batches(batch_size=batch_size)
     return Dataset(pa_Table.from_batches(batch))