Add frozendict as a requirement

google-deepmind · Dec 24, 2021 · de63716 · de63716
1 parent 175ea28
commit de63716
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/setup.py b/setup.py
@@ -48,7 +48,8 @@ def _get_version():
         'numpy>=1.18.0',
     ],
     extras_require={'examples': ['dm-haiku>=0.0.2', 'absl-py>=0.9',
-                                 'optax>=0.0.1', 'scipy>=1.2.1'],
+                                 'frozendict>=2.0.2', 'optax>=0.0.1',
+                                 'scipy>=1.2.1'],
                     'ogb_examples': ['dm-haiku>=0.0.2', 'absl-py>=0.9',
                                      'optax>=0.0.1', 'pandas>=1.0.5',
                                      'dm-tree>=0.1.5']},