From 31dddac7e2c9d6af96efb3e690024b8270a09fe2 Mon Sep 17 00:00:00 2001
From: LiangQuan <lq262469@alibaba-inc.com>
Date: Thu, 11 Jul 2024 13:40:41 +0800
Subject: [PATCH] add RawStorageConfig used for eas storage configuration

---
 pai/model/__init__.py | 2 ++
 pai/model/_model.py   | 8 ++++++++
 2 files changed, 10 insertions(+)

diff --git a/pai/model/__init__.py b/pai/model/__init__.py
index 44892e3..62e551e 100644
--- 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
index 2a349dc..b95966e 100644
--- 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."""