Keep import meltingpot.python working.

PiperOrigin-RevId: 550584032 Change-Id: Id019bb236d8b07bdf2bd48951bf042839cef77c2
google-deepmind · Jul 24, 2023 · a95129b · a95129b
1 parent 2387615
commit a95129b
Showing 1 changed file with 9 additions and 0 deletions.
diff --git a/meltingpot/__init__.py b/meltingpot/__init__.py
@@ -13,6 +13,15 @@
 # limitations under the License.
 """Melting Pot."""
 
+import sys
+
 from meltingpot import bot
 from meltingpot import scenario
 from meltingpot import substrate
+
+try:
+  # Keep `import meltingpot.python` working for external meltingpot.
+  # TODO: b/292470900 - Remove in v3.0
+  sys.modules['meltingpot.python'] = sys.modules['meltingpot']
+except KeyError:
+  pass  # Internal version of meltingpot.