triggerflow 0.2.3__py3-none-any.whl → 0.2.5__py3-none-any.whl

triggerflow/core.py

@@ -458,8 +458,7 @@ class TriggerModel:
             )
 
         
-        if shutil.which("vivado") is not None:
-            build_ugt_model(
+        build_ugt_model(
                 templates_dir=self.subsystem_cfg.get("templates_dir", Path("templates")),
                 firmware_dir=self.workspace_manager.workspace / "firmware",
                 compiler = self.compiler,
@@ -478,12 +477,7 @@ class TriggerModel:
                 object_features=self.subsystem_cfg.get("object_features"),
                 global_features=self.subsystem_cfg.get("global_features")
             )
-        else:
-            warnings.warn(
-                "Vivado executable not found on the system PATH. "
-                "Skipping FW build. ",
-                UserWarning
-            )
+  
 
 
         self.workspace_manager.add_artifact("firmware", self.workspace_manager.workspace / "firmware")

triggerflow/interfaces/uGT.py

@@ -1,12 +1,15 @@
 from pathlib import Path
-import shutil
+import shutil, warnings
 import pkg_resources
 from jinja2 import Template
 import re
 
-def _render_template(template_file: str, output_file: Path, context: dict):
-    with open(template_file, "r") as f:
-        template_text = f.read()
+
+def _render_template(template_name: str, output_file: Path, context: dict):
+    template_bytes = pkg_resources.resource_string(
+        "triggerflow" , template_name
+    )
+    template_text = template_bytes.decode('utf-8')
 
     template = Template(template_text)
     rendered = template.render(**context)
@@ -108,20 +111,29 @@ def build_ugt_model(
 
     out_path = firmware_dir / "firmware/model-gt.cpp"
     
-    _render_template(f"{templates_dir}/model-gt.cpp", out_path, context)
+    _render_template("templates/model-gt.cpp", out_path, context)
 
     out_path = firmware_dir / "firmware/build_ugt.tcl"
-    _render_template(f"{templates_dir}/build_ugt.tcl", out_path, context_tcl)
+    _render_template("templates/build_ugt.tcl", out_path, context_tcl)
 
-    shutil.copy(f"{templates_dir}/data_types.h", firmware_dir / "firmware")
- 
-    
-    subprocess.run(
-        ["vitis_hls", "-f", "build_ugt.tcl"],
-        cwd=firmware_dir/"firmware",
-        check=True
+    shutil.copy(
+        pkg_resources.resource_filename("triggerflow", "templates/data_types.h"), 
+        firmware_dir / "firmware"
     )
 
+    if shutil.which("vivado") is not None:
+        subprocess.run(
+            ["vitis_hls", "-f", "build_ugt.tcl"],
+            cwd=firmware_dir/"firmware",
+            check=True
+        )
+    else:
+            warnings.warn(
+                "Vivado executable not found on the system PATH. "
+                "Skipping FW build. ",
+                UserWarning
+            )
+
 
 
 

{triggerflow-0.2.3.dist-info → triggerflow-0.2.5.dist-info}/METADATA

@@ -1,6 +1,6 @@
 Metadata-Version: 2.4
 Name: triggerflow
-Version: 0.2.3
+Version: 0.2.5
 Summary: Utilities for ML models targeting hardware triggers
 Classifier: Programming Language :: Python :: 3
 Classifier: License :: OSI Approved :: MIT License
@@ -17,6 +17,32 @@ Requires-Dist: pytest-cov~=3.0; extra == "dev"
 Requires-Dist: pytest-mock<2.0,>=1.7.1; extra == "dev"
 Requires-Dist: pytest~=7.2; extra == "dev"
 Requires-Dist: ruff~=0.1.8; extra == "dev"
+Provides-Extra: extended
+Requires-Dist: coffea; extra == "extended"
+Requires-Dist: dask; extra == "extended"
+Requires-Dist: pyarrow; extra == "extended"
+Requires-Dist: pandas==2.0.3; extra == "extended"
+Requires-Dist: uproot4==4.0.0; extra == "extended"
+Requires-Dist: pyarrow==17.0.0; extra == "extended"
+Requires-Dist: hist==2.7.3; extra == "extended"
+Requires-Dist: mlflow==2.21.0; extra == "extended"
+Requires-Dist: tensorflow==2.13.0; extra == "extended"
+Requires-Dist: QKeras==0.9.0; extra == "extended"
+Requires-Dist: keras==2.13.1; extra == "extended"
+Requires-Dist: numpy; extra == "extended"
+Requires-Dist: pyparsing; extra == "extended"
+Requires-Dist: PyYAML; extra == "extended"
+Requires-Dist: matplotlib; extra == "extended"
+Requires-Dist: mplhep; extra == "extended"
+Requires-Dist: h5py; extra == "extended"
+Requires-Dist: xxhash; extra == "extended"
+Requires-Dist: shap; extra == "extended"
+Requires-Dist: awkward-pandas; extra == "extended"
+Requires-Dist: qonnx==0.4.0; extra == "extended"
+Requires-Dist: tf_keras; extra == "extended"
+Requires-Dist: tf2onnx; extra == "extended"
+Requires-Dist: hls4ml; extra == "extended"
+Requires-Dist: conifer; extra == "extended"
 
 # Machine Learning for Hardware Triggers
 

{triggerflow-0.2.3.dist-info → triggerflow-0.2.5.dist-info}/RECORD

@@ -6,10 +6,10 @@ trigger_loader/loader.py,sha256=wMkeZ3k36wpxt-B8OpKOa6j7z0-fnJUqQ-5AbVjNpBM,5158
 trigger_loader/processor.py,sha256=cvBfYmvcr4FLzOHgGE50oy7EkFzFaV80Z_66amqfsEY,7724
 triggerflow/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 triggerflow/cli.py,sha256=ZNQb3XQN8Ir6Hp6KX_ugec9bm2kqxLNZ0KdVGJmnqFA,4498
-triggerflow/core.py,sha256=QMU_zuWrYXuZekB7z8Q90Cuaga3B_uuDnYOPthQGdps,22465
+triggerflow/core.py,sha256=WlIYCKzd2NQJlr8Rw2t9zFvvNPZUlWMEIgkU-ov2E3E,22228
 triggerflow/mlflow_wrapper.py,sha256=yCaIS-H7oC2KxnExj24ka9ylF4A1wgzRIpc7Y43ervI,10667
 triggerflow/interfaces/__init__.py,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
-triggerflow/interfaces/uGT.py,sha256=UBCK0WtF-MUkI9mWrulOdExtDXgwKPsHDS4C-FXIgMs,3637
+triggerflow/interfaces/uGT.py,sha256=aPIYorMrISie1-BSUr-ywWF5AvHruGhGx4ibg21sCMs,4014
 triggerflow/starter/.gitignore,sha256=tH2z_M-tPM9MLWC2wPz1Z43Dq-wdVmb_kVYtrLT3tN4,2052
 triggerflow/starter/README.md,sha256=47DEQpj8HBSa-_TImW-5JCeuQeRkm5NMpJWZG3hSuFU,0
 triggerflow/starter/cookiecutter.json,sha256=neplzXvVe3g6OVqYD8M2olfmjRKNr1UKqmflzP3j1UY,259
@@ -95,8 +95,8 @@ triggerflow/templates/makefile_version,sha256=6kFc_u2oiM9l2rH7RK_BLzdZu1ZEK8PQTQ
 triggerflow/templates/model-gt.cpp,sha256=qZwuTtsvrKB_mOB-HDb2uOD7mDo4-20EjFiQzRjMdPo,2969
 triggerflow/templates/model_template.cpp,sha256=jMNRcO7NgC6I9Wd2BV3Bim-P1qPsAl_oeVQ8KofQGEw,1807
 triggerflow/templates/scales.h,sha256=MFcB5S0DEvfzHuUhyZqILR0O4ktugOG-fLnuCDUUewM,373
-triggerflow-0.2.3.dist-info/METADATA,sha256=8FMEii-KeQD1cm7c2ZrpHI-4f-efE9DmrlP4H9VzmxQ,4353
-triggerflow-0.2.3.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
-triggerflow-0.2.3.dist-info/entry_points.txt,sha256=5QSV9YDseB_FqgVh9q10BdL4b1I6t68rGwPLXgVL60g,53
-triggerflow-0.2.3.dist-info/top_level.txt,sha256=cX0jkuM9tfxGp002ZBQ1AYgx-6D_NgBtomgPL0WA9bE,43
-triggerflow-0.2.3.dist-info/RECORD,,
+triggerflow-0.2.5.dist-info/METADATA,sha256=_psSoh10cZrBvYTjQIKQVw8m4VohIVsjdxlN1MYQazY,5535
+triggerflow-0.2.5.dist-info/WHEEL,sha256=_zCd3N1l69ArxyTb8rzEoP9TpbYXkqRFSNOD5OuxnTs,91
+triggerflow-0.2.5.dist-info/entry_points.txt,sha256=5QSV9YDseB_FqgVh9q10BdL4b1I6t68rGwPLXgVL60g,53
+triggerflow-0.2.5.dist-info/top_level.txt,sha256=cX0jkuM9tfxGp002ZBQ1AYgx-6D_NgBtomgPL0WA9bE,43
+triggerflow-0.2.5.dist-info/RECORD,,