Merge pull request fastmachinelearning#972 from calad0i/vitis_tarball…

…_fix vitis backend tarball fix
calad0i · Feb 21, 2024 · 9db4237 · 9db4237
2 parents cafbda8 + 1c276d7
commit 9db4237
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)