Fix (export): wrap export method not export_onnx

Xilinx · May 5, 2021 · b4390db · b4390db
1 parent 3e24be8
commit b4390db
Showing 1 changed file with 4 additions and 4 deletions.
diff --git a/src/brevitas/export/__init__.py b/src/brevitas/export/__init__.py
@@ -10,19 +10,19 @@
 
 
 def export_finn_onnx(*args, **kwargs):
-    return FINNManager.export_onnx(*args, **kwargs)
+    return FINNManager.export(*args, **kwargs)
 
 
 def export_dpuv1_onnx(*args, **kwargs):
-    return DPUv1Manager.export_onnx(*args, **kwargs)
+    return DPUv1Manager.export(*args, **kwargs)
 
 
 def export_dpuv2_onnx(*args, **kwargs):
-    return DPUv2Manager.export_onnx(*args, **kwargs)
+    return DPUv2Manager.export(*args, **kwargs)
 
 
 def export_standard_onnx(*args, **kwargs):
-    return StdONNXManager.export_onnx(*args, **kwargs)
+    return StdONNXManager.export(*args, **kwargs)
 
 
 def jit_trace_dpuv1(*args, **kwargs):