Merge pull request #9 from liamdugan/bugfix

Fixed bug in `load_data`
liamdugan · Sep 4, 2024 · 1925e93 · 1925e93
2 parents 3f0b9af + 4cfa7ff
commit 1925e93
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/raid/_version.py b/raid/_version.py
@@ -1 +1 @@
-__version__ = "0.0.5"
+__version__ = "0.0.6"
diff --git a/raid/utils.py b/raid/utils.py
@@ -42,10 +42,12 @@ def load_data(split: Literal["train", "test", "extra"], include_adversarial: boo
     if split not in ("train", "test", "extra"):
         raise ValueError('`split` must be one of ("train", "test", "extra")')
 
+    fname = f"{split}.csv" if include_adversarial else f"{split}_none.csv"
+
     if fp is None:
-        fname = f"{split}.csv" if include_adversarial else f"{split}_none.csv"
         fp = RAID_CACHE_DIR / fname
     else:
         fp = Path(fp)
+
     fp = download_file(f"{RAID_DATA_URL_BASE}/{fname}", fp)
     return pd.read_csv(fp)