Merge pull request #21 from yuchen0cc/main

expose OssClient
aliyun · Dec 23, 2024 · 21b8830 · 21b8830
2 parents cf262a7 + 1ff30c1
commit 21b8830
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/oss-torch-connector/osstorchconnector/__init__.py b/oss-torch-connector/osstorchconnector/__init__.py
@@ -1,13 +1,15 @@
 from .oss_iterable_dataset import OssIterableDataset
 from .oss_map_dataset import OssMapDataset
 from .oss_checkpoint import OssCheckpoint
+from ._oss_client import OssClient
 from ._oss_bucket_iterable import imagenet_manifest_parser
 from ._oss_tar_iterable import generate_tar_archive
 
 __all__ = [
     "OssIterableDataset",
     "OssMapDataset",
     "OssCheckpoint",
+    "OssClient",
     "imagenet_manifest_parser",
     "generate_tar_archive",
 ]