Put script under model dir so that it would be used as an entrypoint.

Author: nadiaya

test-toolkit/integration/__init__.py

@@ -29,7 +29,7 @@
 model_gpu_dir = os.path.join(mnist_path, gpu_sub_dir)
 model_eia_dir = os.path.join(mnist_path, eia_sub_dir)
 mnist_eia_script = os.path.join(model_eia_dir, 'mnist.py')
-call_model_fn_once_script = os.path.join(resources_path, 'call_model_fn_once.py')
+call_model_fn_once_script = os.path.join(model_cpu_dir, 'call_model_fn_once.py')
 
 ROLE = 'dummy/unused-role'
 DEFAULT_TIMEOUT = 20

test-toolkit/resources/call_model_fn_once.py renamed to test-toolkit/resources/mnist/model_cpu/call_model_fn_once.py

@@ -14,8 +14,28 @@
 
 import os
 
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
+print("HEY")
 
 def model_fn(model_dir):
+    print("HEY")
+    print("HEY")
+
     lock_file = os.path.join(model_dir, 'model_fn.lock.{}'.format(os.getpid()))
     if os.path.exists(lock_file):
         raise RuntimeError('model_fn called more than once (lock: {})'.format(lock_file))