vitis backend tarball fix

calad0i · Feb 21, 2024 · 1c276d7 · 1c276d7
1 parent c601a59
commit 1c276d7
Showing 1 changed file with 2 additions and 0 deletions.
diff --git a/hls4ml/writer/vitis_writer.py b/hls4ml/writer/vitis_writer.py
@@ -30,3 +30,5 @@ def write_hls(self, model):
         """
         super().write_hls(model)
         self.write_nnet_utils_overrides(model)
+        os.remove(model.config.get_output_dir() + '.tar.gz')
+        self.write_tar(model)