Make LightningLocalFileSystem protected

Lightning-AI · Apr 6, 2021 · 401c344 · 401c344
1 parent 252ef26
commit 401c344
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/pytorch_lightning/utilities/cloud_io.py b/pytorch_lightning/utilities/cloud_io.py
@@ -24,7 +24,7 @@
 import torch
 
 
-class LightningLocalFileSystem(LocalFileSystem):
+class _LightningLocalFileSystem(LocalFileSystem):
     """Extension of ``fsspec.implementations.local.LocalFileSystem`` where ``LightningLocalFileSystem.isdir`` behaves
     the same as ``os.isdir``.
 
@@ -53,7 +53,7 @@ def get_filesystem(path: Union[str, Path]):
         return fsspec.filesystem(path.split(":", 1)[0])
     else:
         # use local filesystem
-        return LightningLocalFileSystem()
+        return _LightningLocalFileSystem()
 
 
 def atomic_save(checkpoint, filepath: str):