File tree Expand file tree Collapse file tree 3 files changed +9
-7
lines changed
Expand file tree Collapse file tree 3 files changed +9
-7
lines changed Original file line number Diff line number Diff line change @@ -26,7 +26,9 @@ def data_path(
2626 filename : str , raise_missing : bool = True , cache_dir : str | None = None
2727) -> str :
2828 if remote_files .is_known_remote (filename ):
29- return remote_files .remote_file (filename , raise_missing = raise_missing )
29+ return remote_files .remote_file (
30+ filename , cache_dir = cache_dir , raise_missing = raise_missing
31+ )
3032
3133 if filename not in known_files and raise_missing :
3234 raise FileNotFoundError (filename )
Original file line number Diff line number Diff line change @@ -72,9 +72,9 @@ def fetch_remote_dataset(
7272 logging .warning ("Downloading %s" , url )
7373 urlretrieve (url , str (writefile ))
7474
75- if tarfile .is_tarfile (writefile ):
75+ if tarfile .is_tarfile (str ( writefile ) ):
7676 logging .warning ("Extracting %s" , writefile )
77- with tarfile .open (writefile ) as tar :
77+ with tarfile .open (str ( writefile ) ) as tar :
7878 members = [tar .getmember (f ) for f in files .values ()]
7979 tar .extractall (str (dataset_dir ), members )
8080
Original file line number Diff line number Diff line change @@ -32,10 +32,10 @@ def test_data_path_cached():
3232
3333
3434def test_delegate_to_remote (monkeypatch , tmpdir ):
35- def dummy_remote_file (filename , data_dir = None , raise_missing = False ):
36- if not data_dir :
37- data_dir = str (tmpdir )
38- return str (Path (data_dir ) / filename )
35+ def dummy_remote_file (filename , cache_dir = None , raise_missing = False ):
36+ if not cache_dir :
37+ cache_dir = str (tmpdir )
38+ return str (Path (cache_dir ) / filename )
3939
4040 monkeypatch .setattr (skhtd .remote_files , "remote_file" , dummy_remote_file )
4141 monkeypatch .setattr (skhtd .remote_files , "is_known_remote" , lambda _ : True )
You can’t perform that action at this time.
0 commit comments