Fix print import

danijar · Jun 24, 2024 · 8a06f50 · 8a06f50
1 parent 68c60ea
commit 8a06f50
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/zerofun/__init__.py b/zerofun/__init__.py
@@ -1,4 +1,4 @@
-__version__ = '2.1.0'
+__version__ = '2.1.1'
 
 import multiprocessing as mp
 try:

diff --git a/zerofun/utils.py b/zerofun/utils.py
@@ -6,7 +6,7 @@
 import time
 import traceback
 
-import embodied
+import elements
 import numpy as np
 import psutil
 
@@ -155,7 +155,7 @@ def warn_remote_error(e, name, lock=get_print_lock):
   msg += 'Call check() to reraise in main process. '
   msg += f'Worker stack trace:\n{full}'
   with lock:
-    embodied.print(msg, color='red')
+    elements.print(msg, color='red')
   if hasattr(e, 'add_note'):
     e.add_note(f'\nWorker stack trace:\n\n{full}')