Merge pull request #13 from dpeerlab/formatting

Format with black and ruff wassersteinwormhole package
dpeerlab · Apr 26, 2024 · 851e4f2 · 851e4f2
2 parents 4cd05d4 + e4ce454
commit 851e4f2
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: