add RawStorageConfig used for eas storage configuration

aliyun · Jul 11, 2024 · 31dddac · 31dddac
1 parent 63b8191
commit 31dddac
Show file tree

Hide file tree

Showing 2 changed files with 10 additions and 0 deletions.
diff --git a/pai/model/__init__.py b/pai/model/__init__.py
@@ -20,6 +20,7 @@
     NfsStorageConfig,
     NodeStorageConfig,
     OssStorageConfig,
+    RawStorageConfig,
     RegisteredModel,
     ResourceConfig,
     SharedMemoryConfig,
@@ -43,4 +44,5 @@
     "NodeStorageConfig",
     "SharedMemoryConfig",
     "OssStorageConfig",
+    "RawStorageConfig",
 ]
diff --git a/pai/model/_model.py b/pai/model/_model.py
@@ -83,6 +83,14 @@ def to_dict(self):
         pass
 
 
+class RawStorageConfig(StorageConfigBase):
+    def __init__(self, config: Dict[str, Any]):
+        self.config = config
+
+    def to_dict(self):
+        return self.config
+
+
 class OssStorageConfig(StorageConfigBase):
     """Configuration for OSS Storage."""