Add optional Numba dependency for faster world generation.

analysis/common.py

@@ -46,6 +46,9 @@ def compute_success_rates(runs, budget=1e6, sortby=None):
 
 def compute_scores(percents):
   # Geometric mean with an offset of 1%.
+  assert all(0 <= x <= 100 for x in percents)
+  if all(x <= 1.0 for x in percents):
+    print('Warning: The input may not be in the right range.')
   with warnings.catch_warnings():  # Empty seeds become NaN.
     warnings.simplefilter('ignore', category=RuntimeWarning)
     scores = np.exp(np.nanmean(np.log(1 + percents), -1)) - 1

setup.py

@@ -4,7 +4,7 @@
 
 setuptools.setup(
     name='crafter',
-    version='1.7.1',
+    version='1.8.0',
     description='Open world survival game for reinforcement learning.',
     url='http://github.com/danijar/crafter',
     long_description=pathlib.Path('README.md').read_text(),
@@ -13,7 +13,11 @@
     package_data={'crafter': ['data.yaml', 'assets/*']},
     entry_points={'console_scripts': ['crafter=crafter.run_gui:main']},
     install_requires=[
-        'numpy', 'imageio', 'pillow', 'opensimplex', 'ruamel.yaml'],
+        'numpy', 'imageio', 'pillow', 'opensimplex', 'ruamel.yaml',
+        # Numba is an optional dependency but we want it installed by default
+        # because it speeds up world generation by ~5x.
+        'numba',
+    ],
     extras_require={'gui': ['pygame']},
     classifiers=[
         'Intended Audience :: Science/Research',