Use Python functions to prep temp directory

rapidsai · Jul 31, 2023 · c8ac39a · c8ac39a
1 parent 07831ef
commit c8ac39a
Showing 1 changed file with 3 additions and 2 deletions.
diff --git a/python/cuml/benchmark/nvtx_benchmark.py b/python/cuml/benchmark/nvtx_benchmark.py
@@ -15,6 +15,7 @@
 #
 
 import os
+import shutil
 import sys
 import tempfile
 from subprocess import run
@@ -26,8 +27,8 @@ def __init__(self, tmp_path=None):
         self.tmp_dir = tempfile.TemporaryDirectory(dir=tmp_path)
         self.nsys_file = os.path.join(self.tmp_dir.name, "report.nsys-rep")
         self.json_file = os.path.join(self.tmp_dir.name, "report.json")
-        self._execute(["rm", "-rf", self.tmp_dir.name])
-        self._execute(["mkdir", "-p", self.tmp_dir.name])
+        shutil.rmtree(self.tmp_dir.name)
+        os.makedirs(self.tmp_dir.name)
 
     def __del__(self):
         self.tmp_dir.cleanup()