Fix bugs in model import.

dohlee · dohlee · commit 58fccb77f8d7 · 2023-02-20T14:01:08.000+09:00
diff --git a/README.md b/README.md
@@ -53,7 +53,7 @@ python create_dataset.py test 0      # ~1 minute, creates dataset_test_0.h5 (0.5
 $ python -m spliceai_pytorch.train --model 80nt \  # 80nt, 400nt, 2k, 10k
   --train-h5 spliceai_train_code/Canonical/dataset_train_all.h5 \
   --test-h5 spliceai_train_code/Canonical/dataset_test_0.h5 \
-  --use-wandb
+  --use-wandb  # Optional, for logging.
 ```
 
 ## Reproduction status (wip)
diff --git a/spliceai_pytorch/__init__.py b/spliceai_pytorch/__init__.py
@@ -1 +1 @@
-from spliceai_pytorch import SpliceAI_80nt, SpliceAI_400nt, SpliceAI_2k, SpliceAI_10k
+from spliceai_pytorch import SpliceAI

Original file line number	Diff line number	Diff line change
`@@ -1 +1 @@`
`1`		`-from spliceai_pytorch import SpliceAI_80nt, SpliceAI_400nt, SpliceAI_2k, SpliceAI_10k`
	`1`	`+from spliceai_pytorch import SpliceAI`