cast list to str

graphnet-team · Sep 14, 2024 · 17d3d44 · 17d3d44
1 parent ea39d1c
commit 17d3d44
Showing 1 changed file with 4 additions and 0 deletions.
diff --git a/src/graphnet/data/curated_datamodule.py b/src/graphnet/data/curated_datamodule.py
@@ -404,6 +404,10 @@ def _prepare_args(
         """
         if backend == "sqlite":
             dataset_path = glob(os.path.join(self.dataset_dir, "*.db"))
+            # Cast from list to string if just 1 path
+            if isinstance(dataset_path, list) & len(dataset_path) == 1:
+                dataset_path: str = dataset_path[0]  # type: ignore
+
             if self._mode == "train":
                 train_val = pd.read_parquet(
                     os.path.join(