Merge pull request #45 from zaxioms/nb-fix

nb fix
google · Feb 9, 2022 · c6acc68 · c6acc68
2 parents 7f869d0 + 97ed16b
commit c6acc68
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/nerfies/configs.py b/nerfies/configs.py
@@ -16,7 +16,7 @@
 from typing import Any, Mapping, Optional, Tuple
 
 import dataclasses
-from flax import nn
+from flax.linen import nn
 import gin
 import immutabledict