File tree Expand file tree Collapse file tree 2 files changed +1
-16
lines changed
checkpoint/orbax/checkpoint/_src/path Expand file tree Collapse file tree 2 files changed +1
-16
lines changed Original file line number Diff line number Diff line change 2424from etils import epath
2525
2626
27- from orbax .checkpoint ._src .path import gcs_utils
28-
2927
3028def get_storage_type (path : epath .Path | str ) -> str :
31- """Returns the storage type of the given path."""
32- if isinstance (path , str ):
33- path = epath .Path (path )
34- if gcs_utils .is_gcs_path (path ):
35- return 'gcs'
36- return 'other'
29+ return 'other'
3730
3831
3932class Timer (object ):
Original file line number Diff line number Diff line change @@ -68,14 +68,6 @@ def test_recursively_copy_files_with_skip_paths(self):
6868 self .assertTrue ((dst_dir / 'file4.txt' ).exists ())
6969 self .assertEqual ((dst_dir / 'file4.txt' ).read_text (), 'content4' )
7070
71- @parameterized .named_parameters (
72- ('_gcs_gs' , 'gs://bucket/path' , 'gcs' ),
73- ('_other_local' , '/tmp/local/path' , 'other' ),
74- ('_other_random' , 'random' , 'other' ),
75- )
76- def test_get_storage_type (self , path , expected_storage_type ):
77- self .assertEqual (utils .get_storage_type (path ), expected_storage_type )
78-
7971
8072if __name__ == '__main__' :
8173 absltest .main ()
You can’t perform that action at this time.
0 commit comments