Remove support for legacy initialize

PiperOrigin-RevId: 745444579

Author: Pathways-on-Cloud Team

pathwaysutils/__init__.py

@@ -83,12 +83,14 @@ def initialize() -> None:
   # debug log is not triggered for customers who are following our instructions
   # and using the new initialize() function only once but have already had the
   # legacy initialization triggered.
-  if _initialization_count > 2:
-    _logger.debug("Already initialized. Ignoring duplicate call.")
-
   if _initialization_count > 1:
+    _logger.info("Already initialized. Ignoring duplicate call.")
     return
 
+  _logger.debug(
+      "Starting initialize."
+  )
+
   if is_pathways_backend_used():
     _logger.debug("Detected Pathways-on-Cloud backend. Applying changes.")
     proxy_backend.register_backend_factory()
@@ -112,6 +114,3 @@ def initialize() -> None:
     _logger.debug(
         "Did not detect Pathways-on-Cloud backend. No changes applied."
     )
-
-
-initialize()

pathwaysutils/test/pathwaysutils_test.py

@@ -43,27 +43,24 @@ def test_first_initialize(self):
     with self.assertLogs(pathwaysutils._logger, level="DEBUG") as logs:
       pathwaysutils.initialize()
 
-    self.assertLen(logs.output, 1)
+    self.assertLen(logs.output, 2)
     self.assertIn(
-        "Detected Pathways-on-Cloud backend. Applying changes.", logs.output[0]
+        "Starting initialize.", logs.output[0]
+    )
+    self.assertIn(
+        "Detected Pathways-on-Cloud backend. Applying changes.", logs.output[1]
     )
-
-  def test_second_initialize(self):
-    jax.config.update("jax_platforms", "proxy")
-    pathwaysutils._initialization_count = 1
-
-    with self.assertNoLogs(pathwaysutils._logger, level="DEBUG"):
-      pathwaysutils.initialize()
 
   @parameterized.named_parameters(
+      ("initialization_count 1", 1),
       ("initialization_count 2", 2),
       ("initialization_count 5", 5),
       ("initialization_count 1000", 1000),
   )
-  def test_initialize_more_than_twice(self, initialization_count):
+  def test_initialize_more_than_once(self, initialization_count):
     pathwaysutils._initialization_count = initialization_count
 
-    with self.assertLogs(pathwaysutils._logger, level="DEBUG") as logs:
+    with self.assertLogs(pathwaysutils._logger, level="INFO") as logs:
       pathwaysutils.initialize()
 
     self.assertLen(logs.output, 1)
@@ -107,4 +104,4 @@ def test_persistence_enabled(self):
 
 
 if __name__ == "__main__":
-  absltest.main()
+  absltest.main()