Skip to content

Commit c6b4b10

Browse files
author
Orbax Authors
committed
No public description
PiperOrigin-RevId: 826777955
1 parent 328bbeb commit c6b4b10

File tree

2 files changed

+1
-16
lines changed

2 files changed

+1
-16
lines changed

checkpoint/orbax/checkpoint/_src/path/utils.py

Lines changed: 1 addition & 8 deletions
Original file line numberDiff line numberDiff line change
@@ -24,16 +24,9 @@
2424
from etils import epath
2525

2626

27-
from orbax.checkpoint._src.path import gcs_utils
28-
2927

3028
def 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

3932
class Timer(object):

checkpoint/orbax/checkpoint/_src/path/utils_test.py

Lines changed: 0 additions & 8 deletions
Original file line numberDiff line numberDiff 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

8072
if __name__ == '__main__':
8173
absltest.main()

0 commit comments

Comments
 (0)