haystack-ml-stack 0.2.0__tar.gz → 0.2.2__tar.gz

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.2.0
+Version: 0.2.2
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/pyproject.toml

@@ -5,7 +5,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "haystack-ml-stack"
-version = "0.2.0"
+version = "0.2.2"
 description = "Functions related to Haystack ML"
 readme = "README.md"
 authors = [{ name = "Oscar Vega", email = "oscar@haystack.tv" }]

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/src/haystack_ml_stack/__init__.py

@@ -1,4 +1,4 @@
 from .app import create_app
 
 __all__ = ["create_app"]
-__version__ = "0.2.0"
+__version__ = "0.2.2"

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/src/haystack_ml_stack/app.py

@@ -143,16 +143,15 @@ def create_app(
         # Synchronous model execution (user code)
         try:
             preprocess_start = time.perf_counter_ns()
+            model["params"]["query_params"] = query_params
             model_input = model["preprocess"](
                 user,
                 streams,
                 playlist,
-                {**model.get("params"), "query_params": query_params},
+                model["params"],
             )
             predict_start = time.perf_counter_ns()
-            model_output = model["predict"](
-                model_input, {**model.get("params"), "query_params": query_params}
-            )
+            model_output = model["predict"](model_input, model["params"])
             predict_end = time.perf_counter_ns()
         except Exception as e:
             logger.error("Model prediction failed: %s", e)

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/src/haystack_ml_stack/dynamo.py

@@ -169,7 +169,9 @@ async def set_stream_features(
         features_cache[cache_key] = {
             "value": parsed.get("value"),
             "cache_ttl_in_seconds": int(parsed.get("cache_ttl_in_seconds", -1)),
-            "updated_at": datetime.datetime.fromisoformat(parsed.get("updated_at")),
+            "updated_at": datetime.datetime.fromisoformat(
+                parsed.get("updated_at")
+            ).replace(tzinfo=None),
         }
         if cache_key in cache_miss:
             cache_miss[cache_key][feature_name] = parsed.get("value")

{haystack_ml_stack-0.2.0 → haystack_ml_stack-0.2.2}/src/haystack_ml_stack.egg-info/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: haystack-ml-stack
-Version: 0.2.0
+Version: 0.2.2
 Summary: Functions related to Haystack ML
 Author-email: Oscar Vega <oscar@haystack.tv>
 License: MIT