Replace deprecated load_module()

Signed-off-by: coleramos425 <[email protected]>
ROCm · Nov 21, 2024 · a180f92 · a180f92
1 parent 1fea7f8
commit a180f92
Show file tree

Hide file tree

Showing 4 changed files with 4 additions and 19 deletions.
diff --git a/tests/test_analyze_commands.py b/tests/test_analyze_commands.py
@@ -1,14 +1,13 @@
 import os.path
 import shutil
 from importlib.machinery import SourceFileLoader
-from pathlib import Path
 from unittest.mock import patch
 
 import pandas as pd
 import pytest
 import test_utils
 
-rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").load_module()
+rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").exec_module()
 
 baseline_opts = ["rocprof-compute", "analyze"]
 

diff --git a/tests/test_analyze_workloads.py b/tests/test_analyze_workloads.py
@@ -1,18 +1,9 @@
-import inspect
-import os.path
-import re
-import shutil
-import subprocess
-import sys
 from importlib.machinery import SourceFileLoader
-from pathlib import Path
 from unittest.mock import patch
 
-import pandas as pd
 import pytest
-import test_utils
 
-rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").load_module()
+rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").exec_module()
 
 ##################################################
 ##          Generated tests                     ##

diff --git a/tests/test_import_workloads.py b/tests/test_import_workloads.py
@@ -1,9 +1,7 @@
 from importlib.machinery import SourceFileLoader
 from unittest.mock import patch
 
-import pytest
-
-rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").load_module()
+rocprof_compute = SourceFileLoader("rocprof-compute", "src/rocprof-compute").exec_module()
 
 ##################################################
 ##          Generated tests                     ##

diff --git a/tests/test_profile_general.py b/tests/test_profile_general.py
@@ -1,12 +1,9 @@
 import inspect
 import os.path
 import re
-import shutil
 import subprocess
 import sys
 from importlib.machinery import SourceFileLoader
-from pathlib import Path
-from unittest.mock import patch
 
 import pandas as pd
 import pytest
@@ -37,7 +34,7 @@
 config = {}
 config["rocprofiler-compute"] = SourceFileLoader(
     "rocprofiler-compute", "src/rocprof-compute"
-).load_module()
+).exec_module()
 config["kernel_name_1"] = "vecCopy(double*, double*, double*, int, int) [clone .kd]"
 config["app_1"] = ["./tests/vcopy", "-n", "1048576", "-b", "256", "-i", "3"]
 config["cleanup"] = True