progress

lib/pt_strategy/pt_model.py

@@ -3,6 +3,9 @@ from __future__ import annotations
 from abc import ABC, abstractmethod
 from typing import Any, Dict, cast
 
+# ---
+from cvttpy_tools.config import Config
+# ---
 from pairs_trading.lib.pt_strategy.prediction import Prediction
 from pairs_trading.lib.pt_strategy.trading_pair import TradingPair
     
@@ -13,10 +16,10 @@ class PairsTradingModel(ABC):
         ...
     
     @staticmethod
-    def create(config: Dict[str, Any]) -> PairsTradingModel:
+    def create(config: Config) -> PairsTradingModel:
         import importlib
         
-        model_class_name = config.get("model_class", None)
+        model_class_name = config.get_value("model_class", None)
         assert model_class_name is not None
         module_name, class_name = model_class_name.rsplit(".", 1)
         module = importlib.import_module(module_name)