Merge pull request #24 from SauravMaheshkar:master

PiperOrigin-RevId: 419575158 Change-Id: Id2cf7b8f029eba9b0f6dccc6b45e3835926328aa
google-deepmind · Jan 4, 2022 · 0e02473 · 0e02473
2 parents 175ea28 + de63716
commit 0e02473
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']},