triggerflow 0.2.9__tar.gz → 0.3.0__tar.gz

{triggerflow-0.2.9/src/triggerflow.egg-info → triggerflow-0.3.0}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.2.9
+Version: 0.3.0
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{triggerflow-0.2.9 → triggerflow-0.3.0}/pyproject.toml

@@ -4,7 +4,7 @@ build-backend = "setuptools.build_meta"
 
 [project]
 name = "triggerflow"
-version = "0.2.9"
+version = "0.3.0"
 description = "Utilities for ML models targeting hardware triggers"
 readme = "README.md"
 requires-python = ">=3.11"

{triggerflow-0.2.9 → triggerflow-0.3.0}/src/triggerflow/core.py

@@ -232,9 +232,10 @@ class ConverterFactory:
     @staticmethod
     def create_converter(ml_backend: str, compiler: str) -> ModelConverter:
         if ml_backend.lower() == "keras" and compiler.lower() == "hls4ml":
-            return KerasToQONNXConverter()
-        else:
-            return NoOpConverter()
+            import keras
+            if not keras.__version__.startswith("3"):
+                return KerasToQONNXConverter()
+        return NoOpConverter()
 
 
 class CompilerFactory:
@@ -284,7 +285,7 @@ class WorkspaceManager:
     def save_native_model(self, model, ml_backend: str):
         """Save the native model to workspace"""
         if ml_backend.lower() == "keras":
-            model.save(self.workspace / "keras_model")
+            model.save(self.workspace / "keras_model.h5")
         elif ml_backend.lower() == "xgboost":
             model.save_model(str(self.workspace / "xgb_model.json"))
     
@@ -609,4 +610,4 @@ class TriggerModel:
         obj.converter = ConverterFactory.create_converter(obj.ml_backend, obj.compiler)
         obj.dataset_object = None 
         
-        return obj
+        return obj

{triggerflow-0.2.9 → triggerflow-0.3.0/src/triggerflow.egg-info}/PKG-INFO

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.2.9
+Version: 0.3.0
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License

{triggerflow-0.2.9 → triggerflow-0.3.0}/tests/test.py

@@ -1,4 +1,5 @@
 import numpy as np
+import keras
 from keras.layers import Input
 from keras.models import Model
 from qkeras.qlayers import QActivation, QDense