Remove requirements.txt for GPU local integration tests

Author: sachanub

CHANGELOG.md

@@ -1,5 +1,11 @@
 # Changelog
 
+## v2.0.18 (2023-10-10)
+
+### Bug Fixes and Other Changes
+
+ * Fix integration tests and update Python versions
+
 ## v2.0.17 (2023-08-07)
 
 ### Bug Fixes and Other Changes

VERSION

@@ -1 +1 @@
-2.0.18.dev0
+2.0.19.dev0

test/integration/__init__.py

@@ -45,21 +45,17 @@
 
 model_cpu_1d_dir = os.path.join(model_cpu_dir, '1d')
 mnist_1d_script = os.path.join(model_cpu_1d_dir, code_sub_dir, 'mnist_1d.py')
-mnist_1d_requirements = os.path.join(model_cpu_1d_dir, code_sub_dir, 'requirements.txt')
 model_cpu_1d_tar = file_utils.make_tarfile(mnist_1d_script,
                                            os.path.join(model_cpu_1d_dir, "torch_model.pth"),
                                            model_cpu_1d_dir,
-                                           script_path="code",
-                                           requirements=mnist_1d_requirements)
+                                           script_path="code")
 
 model_gpu_dir = os.path.join(mnist_path, gpu_sub_dir)
 mnist_gpu_script = os.path.join(model_gpu_dir, code_sub_dir, 'mnist.py')
-mnist_gpu_requirements = os.path.join(model_gpu_dir, code_sub_dir, 'requirements.txt')
 model_gpu_tar = file_utils.make_tarfile(mnist_gpu_script,
                                         os.path.join(model_gpu_dir, "torch_model.pth"),
                                         model_gpu_dir,
-                                        script_path="code",
-                                        requirements=mnist_gpu_requirements)
+                                        script_path="code")
 
 model_eia_dir = os.path.join(mnist_path, eia_sub_dir)
 mnist_eia_script = os.path.join(model_eia_dir, 'mnist.py')
@@ -74,13 +70,11 @@
                                              model_inductor_dir)
 
 call_model_fn_once_script = os.path.join(model_cpu_dir, code_sub_dir, 'call_model_fn_once.py')
-call_model_fn_once_requirements = os.path.join(model_cpu_dir, code_sub_dir, 'requirements.txt')
 call_model_fn_once_tar = file_utils.make_tarfile(call_model_fn_once_script,
                                                  os.path.join(model_cpu_dir, "torch_model.pth"),
                                                  model_cpu_dir,
                                                  "model_call_model_fn_once.tar.gz",
-                                                 script_path="code",
-                                                 requirements=call_model_fn_once_requirements)
+                                                 script_path="code")
 
 default_model_dir = os.path.join(resnet18_path, default_sub_dir)
 default_model_script = os.path.join(default_model_dir, code_sub_dir, "resnet18.py")

test/resources/mnist/model_cpu/1d/code/mnist_1d.py

@@ -12,11 +12,6 @@
 # language governing permissions and limitations under the License.
 from __future__ import absolute_import
 
-try:
-    import transformers
-except ImportError:
-    raise ImportError("The 'transformers' module was not found.'")
-    
 import os
 import torch
 import torch.nn as nn

test/resources/mnist/model_cpu/1d/code/requirements.txt

@@ -14,11 +14,6 @@
 
 import os
 
-try:
-    import transformers
-except ImportError:
-    raise ImportError("The 'transformers' module was not found.'")
-    
 def model_fn(model_dir):
     lock_file = os.path.join(model_dir, 'model_fn.lock.{}'.format(os.getpid()))
     if os.path.exists(lock_file):

test/resources/mnist/model_cpu/code/call_model_fn_once.py

@@ -16,11 +16,6 @@
 import os
 import sys
 
-try:
-    import transformers
-except ImportError:
-    raise ImportError("The 'transformers' module was not found.'")
-
 import torch
 import torch.nn as nn
 import torch.nn.functional as F