mainsequence 3.17.2__tar.gz → 3.17.4__tar.gz

{mainsequence-3.17.2 → mainsequence-3.17.4}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 3.17.2
+Version: 3.17.4
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-3.17.2 → mainsequence-3.17.4}/mainsequence/client/models_simple_tables.py

@@ -790,12 +790,18 @@ class SimpleTableUpdate(TableUpdateNode, BaseObjectOrm):
         depth_df = pd.DataFrame(r.json())
 
         if not depth_df.empty:
-            # Normalize legacy backend keys to the current dependency id column.
-            depth_df = depth_df.rename(
-                columns={
-                    "local_time_serie_id": "update_node_id",
-                    "data_node_update_id": "update_node_id",
-                }
+            if "local_time_serie_id" in depth_df.columns and "data_node_update_id" in depth_df.columns:
+                depth_df["update_node_id"] = depth_df["local_time_serie_id"].fillna(
+                    depth_df["data_node_update_id"]
+                )
+            elif "local_time_serie_id" in depth_df.columns:
+                depth_df = depth_df.rename(columns={"local_time_serie_id": "update_node_id"})
+            elif "data_node_update_id" in depth_df.columns:
+                depth_df = depth_df.rename(columns={"data_node_update_id": "update_node_id"})
+
+            depth_df = depth_df.drop(
+                columns=["local_time_serie_id", "data_node_update_id"],
+                errors="ignore",
             )
 
         return depth_df

{mainsequence-3.17.2 → mainsequence-3.17.4}/mainsequence/client/models_tdag.py

@@ -469,16 +469,24 @@ class DataNodeUpdate(TableUpdateNode, BaseObjectOrm):
             raise Exception(f"Error in request {r.text}")
 
         depth_df = pd.DataFrame(r.json())
-
         if not depth_df.empty:
             # Normalize legacy backend keys to the current dependency id column.
-            depth_df = depth_df.rename(
-                columns={
-                    "local_time_serie_id": "update_node_id",
-                    "data_node_update_id": "update_node_id",
-                }
+            if "local_time_serie_id" in depth_df.columns and "data_node_update_id" in depth_df.columns:
+                depth_df["update_node_id"] = depth_df["local_time_serie_id"].fillna(
+                    depth_df["data_node_update_id"]
+                )
+            elif "local_time_serie_id" in depth_df.columns:
+                depth_df = depth_df.rename(columns={"local_time_serie_id": "update_node_id"})
+            elif "data_node_update_id" in depth_df.columns:
+                depth_df = depth_df.rename(columns={"data_node_update_id": "update_node_id"})
+
+            depth_df = depth_df.drop(
+                columns=["local_time_serie_id", "data_node_update_id"],
+                errors="ignore",
             )
 
+
+
         return depth_df
 
     @classmethod

{mainsequence-3.17.2 → mainsequence-3.17.4}/mainsequence/tdag/data_nodes/data_nodes.py

@@ -36,6 +36,7 @@ from mainsequence.client.models_tdag import (
 )
 from mainsequence.instrumentation import tracer
 from mainsequence.logconf import logger
+from mainsequence.tdag.base_persist_managers import get_data_node_source_code
 from mainsequence.tdag.config import ogm
 from mainsequence.tdag.data_nodes.persist_managers import APIPersistManager, PersistManager
 
@@ -43,7 +44,6 @@ from ..configuration_models import BaseConfiguration
 from .models import DataNodeConfiguration
 from .namespacing import current_hash_namespace
 from .namespacing import hash_namespace as _hash_namespace_cm
-from .persist_managers import get_data_node_source_code
 
 
 def get_data_source_from_orm() -> Any:
@@ -1139,6 +1139,8 @@ class DataNode(DataAccessMixin, ABC):
             self.dependencies_df = depth_df[
                 depth_df["update_node_id"] != self.data_node_update.id
             ].copy()
+
+
         else:
             self.dependencies_df = pd.DataFrame()
 

{mainsequence-3.17.2 → mainsequence-3.17.4}/mainsequence.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: mainsequence
-Version: 3.17.2
+Version: 3.17.4
 Summary: Main Sequence SDK 
 Author-email: Main Sequence GmbH <dev@main-sequence.io>
 License: MainSequence GmbH SDK License Agreement

{mainsequence-3.17.2 → mainsequence-3.17.4}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "mainsequence"
-version = "3.17.2"
+version = "3.17.4"
 description = "Main Sequence SDK "
 readme = "README.md"
 requires-python = ">=3.10"