Skip to content

Commit f1ce079

Browse files
committed
test: disable nonjitworkflow test
1 parent 4fea335 commit f1ce079

File tree

1 file changed

+0
-25
lines changed

1 file changed

+0
-25
lines changed

tests/test_workflows.py

Lines changed: 0 additions & 25 deletions
Original file line numberDiff line numberDiff line change
@@ -90,31 +90,6 @@ def test_std_workflow():
9090
assert min_fitness1 < 1e-4
9191

9292

93-
def test_non_jit_workflow():
94-
monitor = StdSOMonitor()
95-
# create a workflow
96-
workflow = workflows.NonJitWorkflow(
97-
algorithm=algorithms.CSO(
98-
lb=jnp.full(shape=(2,), fill_value=-32),
99-
ub=jnp.full(shape=(2,), fill_value=32),
100-
pop_size=20,
101-
),
102-
problem=problems.numerical.Ackley(),
103-
monitors=[monitor],
104-
)
105-
# init the workflow
106-
key = jax.random.PRNGKey(42)
107-
state = workflow.init(key)
108-
109-
# run the workflow for 100 steps
110-
for i in range(100):
111-
state = workflow.step(state)
112-
113-
# the result should be close to 0
114-
min_fitness = monitor.get_best_fitness()
115-
assert min_fitness < 1e-4
116-
117-
11893
def test_distributed_cso():
11994
monitor = StdSOMonitor()
12095
# create a workflow

0 commit comments

Comments
 (0)