Format with black and ruff wassersteinwormhole package

dpeerlab · Apr 26, 2024 · e4ce454 · e4ce454
1 parent 4cd05d4
commit e4ce454
Show file tree

Hide file tree

Showing 6 changed files with 409 additions and 288 deletions.
diff --git a/wassersteinwormhole/DefaultConfig.py b/wassersteinwormhole/DefaultConfig.py
@@ -2,7 +2,7 @@
 from flax import linen as nn
 import jax.numpy as jnp
 
-from typing import Callable, Any, Optional
+from typing import Callable, Any
 
 @struct.dataclass
 class DefaultConfig: