mv aviary/wrenformer/run.py examples/wrenformer.py

janosh · janosh · commit bd08f56537c3 · 2022-06-16T20:48:24.000+01:00
diff --git a/examples/mat_bench/run_wrenformer.py b/examples/mat_bench/run_wrenformer.py
@@ -9,9 +9,9 @@
 from matbench.task import MatbenchTask
 
 from aviary.core import TaskType
-from aviary.wrenformer.run import run_wrenformer
 from aviary.wrenformer.utils import merge_json_on_disk
 from examples.mat_bench import DATA_PATHS, MODULE_DIR
+from examples.wrenformer import run_wrenformer
 
 __author__ = "Janosh Riebesell"
 __date__ = "2022-04-11"
diff --git a/examples/mp_wbm/run_wrenformer.py b/examples/mp_wbm/run_wrenformer.py
@@ -7,7 +7,7 @@
 import pandas as pd
 
 from aviary import ROOT
-from aviary.wrenformer.run import run_wrenformer
+from examples.wrenformer import run_wrenformer
 
 __author__ = "Janosh Riebesell"
 __date__ = "2022-06-13"
diff --git a/examples/wrenformer.py b/examples/wrenformer.py
diff --git a/tests/test_wrenformer.py b/tests/test_wrenformer.py
@@ -1,4 +1,4 @@
-from aviary.wrenformer.run import run_wrenformer
+from examples.wrenformer import run_wrenformer
 
 
 def test_wrenformer_regression(df_matbench_jdft2d):

Original file line number	Diff line number	Diff line change
`@@ -1,4 +1,4 @@`
`1`		`-from aviary.wrenformer.run import run_wrenformer`
	`1`	`+from examples.wrenformer import run_wrenformer`
`2`	`2`
`3`	`3`
`4`	`4`	`def test_wrenformer_regression(df_matbench_jdft2d):`