Fix AutoBNN's pip package by including all of its dependencies (and not

just the ones it has in addition to tfp's). PiperOrigin-RevId: 620032524
tensorflow · Mar 28, 2024 · ae05b63 · ae05b63
1 parent df46aff
commit ae05b63
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/spinoffs/autobnn/setup.py b/spinoffs/autobnn/setup.py
@@ -46,10 +46,13 @@
         'bayeux-ml',
         'chex',
         'flax',
+        'jax',
         'jaxtyping',
         'matplotlib',
+        'numpy',
         'pandas',
         'scipy',
+        'tensorflow-probability'
         ],
     # Add in any packaged data.
     exclude_package_data={'': ['BUILD']},