Merge pull request #30 from transformerlab/add/config-string-checks

add back string checks for exp config

Author: deep1401

pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "transformerlab"
-version = "0.0.19"
+version = "0.0.20"
 description = "Python SDK for Transformer Lab"
 readme = "README.md"
 requires-python = ">=3.10"

src/lab/experiment.py

@@ -38,23 +38,36 @@ def _initialize(self):
     def update_config_field(self, key, value):
         """Update a single key in config."""
         current_config = self._get_json_data_field("config", {})
+        if isinstance(current_config, str):
+            try:
+                current_config = json.loads(current_config)
+            except json.JSONDecodeError:
+                current_config = {}
         current_config[key] = value
         self._update_json_data_field("config", current_config)
 
     @classmethod
     def create_with_config(cls, name: str, config: dict) -> 'Experiment':
         """Create an experiment with config."""
-        if not isinstance(config, dict):
-            raise TypeError("Config must be a dictionary")
+        if isinstance(config, str):
+            try:
+                config = json.loads(config)
+            except json.JSONDecodeError:
+                raise TypeError("config must be a dict or valid JSON string")
+        elif not isinstance(config, dict):
+            raise TypeError("config must be a dict")
         exp = cls.create(name)
         exp._update_json_data_field("config", config)
         return exp
 
     def update_config(self, config: dict):
         """Update entire config."""
-        if not isinstance(config, dict):
-            raise TypeError("Config must be a dictionary")
         current_config = self._get_json_data_field("config", {})
+        if isinstance(current_config, str):
+            try:
+                current_config = json.loads(current_config)
+            except json.JSONDecodeError:
+                current_config = {}
         current_config.update(config)
         self._update_json_data_field("config", current_config)