ep-sdk-4pd 0.1.7__tar.gz → 0.1.9__tar.gz

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ep_sdk_4pd
-Version: 0.1.7
+Version: 0.1.9
 Summary: 4paradigm Electricity Platform Service SDK Library for Python
 Home-page: https://gitlab.4pd.io/electricityproject/electricity-platform-sdk
 Author: 4paradigm Electricity Platform SDK

ep_sdk_4pd-0.1.9/ep_sdk_4pd/__init__.py

@@ -0,0 +1 @@
+__version__ = '0.1.9'

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/ep_sdk_4pd/ep_data.py

@@ -70,12 +70,18 @@ class EpData:
 
     @staticmethod
     def get_predict_data(
-            scope
+            scope,
+            is_test=False,
+            test_time=None,
     ):
         date_str = EpSystem.get_system_date(is_online=True)
         calculated_date = datetime.strptime(date_str, "%Y-%m-%d")
         system_date = calculated_date.strftime("%Y-%m-%d")  # 转换回字符串
 
+        # 测试
+        if is_test:
+            system_date=test_time
+
         request = PredictDataRequest(
             scope=scope,
             system_date=system_date,

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/ep_sdk_4pd/ep_system.py

@@ -121,18 +121,24 @@ class EpSystem:
             return None
 
     @staticmethod
-    def call_train_done(strategy_id: int = None):
-        if strategy_id is None:
+    def call_train_done(
+            strategy_id: int = None,
+            script_strategy_id: int = None
+    ):
+        if (strategy_id is None
+                or script_strategy_id is None):
             return False
 
-        request = CallTrainDoneRequest(strategy_id=strategy_id)
+        request = CallTrainDoneRequest(strategy_id=strategy_id,
+                                       script_strategy_id=script_strategy_id)
         full_url = f'{endpoint}{request.api}'
         headers = {
             'content-type': request.content_type,
         }
 
         payload = {
-            'strategy_id': request.strategy_id
+            'strategy_id': request.strategy_id,
+            'script_strategy_id': request.script_strategy_id
         }
 
         response = requests.request(
@@ -152,4 +158,4 @@ class EpSystem:
         if response.code == 200:
             return True
         else:
-            return False
+            return False

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/ep_sdk_4pd/models.py

@@ -178,12 +178,18 @@ class CallTrainDoneRequest(BaseRequest):
     输入：无，默认为EpSystem.get_run_strategy()["id"]
     """
 
-    def __init__(self, strategy_id: int = None):
+    def __init__(
+            self,
+            strategy_id: int = None,
+            script_strategy_id: int = None
+    ):
         """
         Args:
-            strategy_id: 是否线上环境
+            strategy_id
+            script_strategy_id
         """
         self.strategy_id = strategy_id
+        self.script_strategy_id = script_strategy_id
 
         super().__init__()
         self.api = f'/ep/api/sdk/call_train_done'

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/ep_sdk_4pd.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.1
 Name: ep_sdk_4pd
-Version: 0.1.7
+Version: 0.1.9
 Summary: 4paradigm Electricity Platform Service SDK Library for Python
 Home-page: https://gitlab.4pd.io/electricityproject/electricity-platform-sdk
 Author: 4paradigm Electricity Platform SDK

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/setup.py

@@ -8,7 +8,7 @@ DESCRIPTION = '4paradigm Electricity Platform Service SDK Library for Python'
 AUTHOR = '4paradigm Electricity Platform SDK'
 AUTHOR_EMAIL = ''
 URL = 'https://gitlab.4pd.io/electricityproject/electricity-platform-sdk'
-VERSION = '0.1.7'
+VERSION = '0.1.9'
 REQUIRES = ['requests']
 
 LONG_DESCRIPTION = ''

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/tests/test_call_train_done.py

@@ -4,7 +4,7 @@ from ep_sdk_4pd.ep_system import EpSystem
 def test_call_train_done():
     print('-------------test_call_train_done-------------')
 
-    data = EpSystem.call_train_done(strategy_id=None)
+    data = EpSystem.call_train_done(strategy_id=None,script_strategy_id=None)
     print(data)
     print('-------------------------------------')
 

{ep_sdk_4pd-0.1.7 → ep_sdk_4pd-0.1.9}/tests/test_predict_data.py

@@ -4,7 +4,7 @@ from ep_sdk_4pd.ep_data import EpData
 def test_predict_data():
     print('-------------test_predict_data-------------')
 
-    data = EpData.get_predict_data(scope="weather")
+    data = EpData.get_predict_data(scope="plant",is_test=True,test_time='2023-04-02')
     print(data)
     print('-------------------------------------')
 

ep_sdk_4pd-0.1.7/ep_sdk_4pd/__init__.py

@@ -1 +0,0 @@
-__version__ = '0.1.7'