Fix CI

Author: federicobrancasi

DeepQuant/__init__.py

@@ -4,6 +4,12 @@
 #
 # Federico Brancasi <[email protected]>
 
+# FBRANCASI: Workaround for PyTorch/FX API change: ensure private alias exists
+import torch.fx.node as _fx_node
+
+if not hasattr(_fx_node.Node, "_Node__update_args_kwargs"):
+    _fx_node.Node._Node__update_args_kwargs = _fx_node.Node._update_args_kwargs
+
 from DeepQuant.Export import brevitasToTrueQuant
 
 __all__ = ["brevitasToTrueQuant"]

pyproject.toml

@@ -33,6 +33,7 @@ dependencies = [
     "onnx",
     "onnxoptimizer",
     "onnxruntime",
+    "ultralytics",
 ]
 
 [tool.setuptools]