adjust change

RasmusOrsoe · Sep 13, 2024 · c0e0f5a · c0e0f5a
1 parent 355a691
commit c0e0f5a
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/src/graphnet/data/extractors/internal/parquet_extractor.py b/src/graphnet/data/extractors/internal/parquet_extractor.py
@@ -29,7 +29,7 @@ def __init__(self, extractor_name: str):
 
     def __call__(self, file_path: str) -> pd.DataFrame:
         """Extract information from parquet file."""
-        if self._table == os.path.dirname(file_path):
+        if self._table == os.path.basename(os.path.dirname(file_path)):
             return pol.read_parquet(file_path).to_pandas()
         else:
             return None