cleanup

Author: JyotinderSingh

tests/e2e/cpu_execution_test.py

@@ -14,14 +14,13 @@
 from absl.testing import absltest
 
 import keras_remote
-from tests.e2e.e2e_utils import get_gcp_project, skip_unless_e2e
+from tests.e2e.e2e_utils import skip_unless_e2e
 
 
 @skip_unless_e2e()
 class TestCpuExecution(absltest.TestCase):
   def setUp(self):
     super().setUp()
-    self.project = get_gcp_project()
 
   def test_simple_function(self):
     """Execute a simple add function remotely and verify the result."""

tests/e2e/e2e_utils.py

@@ -7,11 +7,3 @@
 def skip_unless_e2e(reason="E2E_TESTS not set"):
   """Skip decorator for e2e tests unless E2E_TESTS env var is set."""
   return unittest.skipUnless(os.environ.get("E2E_TESTS"), reason)
-
-
-def get_gcp_project():
-  """Return GCP project from env, skip test if not set."""
-  project = os.environ.get("KERAS_REMOTE_PROJECT")
-  if not project:
-    raise unittest.SkipTest("KERAS_REMOTE_PROJECT not set")
-  return project