[Benchmark-Py] Refactoring & Bug Fix sys.path.insert(...)

Author: DEKHTIARJonathan

tftrt/benchmarking-python/huggingface/bert/transformers.py

@@ -28,8 +28,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-sys.path.insert(0, parentdir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/image_classification/image_classification.py

@@ -30,8 +30,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-sys.path.insert(0, parentdir)
+
+benchmark_base_dir = os.path.dirname(currentdir)
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/bert_tf2/infer.py

@@ -34,9 +34,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/dlrm_tf2/infer.py

@@ -28,9 +28,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI as CommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/efficientnet_base/infer.py

@@ -28,9 +28,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/electra_tf2/infer.py

@@ -38,9 +38,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/mrcnn_tf2/infer.py

@@ -30,9 +30,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/nnunet2d_tf2/infer.py

@@ -36,9 +36,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/nnunet3d_tf2/infer.py

@@ -36,9 +36,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner

tftrt/benchmarking-python/nvidia_examples/sim_tf2/infer.py

@@ -28,9 +28,9 @@
 currentdir = os.path.dirname(
     os.path.abspath(inspect.getfile(inspect.currentframe()))
 )
-parentdir = os.path.dirname(currentdir)
-basedir = os.path.dirname(parentdir)
-sys.path.insert(0, basedir)
+
+benchmark_base_dir = os.path.dirname(os.path.dirname(currentdir))
+sys.path.insert(0, benchmark_base_dir)
 
 from benchmark_args import BaseCommandLineAPI
 from benchmark_runner import BaseBenchmarkRunner