Skip to content

Commit

Permalink
cleaning
Browse files Browse the repository at this point in the history
  • Loading branch information
Kevin CO committed Jul 25, 2024
1 parent 1d0da90 commit d4e1831
Showing 1 changed file with 3 additions and 3 deletions.
6 changes: 3 additions & 3 deletions tests/shard6/test_time_dependent_custom_model.py
Original file line number Diff line number Diff line change
Expand Up @@ -130,7 +130,7 @@ def dynamics(
defects=None,
)

def declare_ding_variables(
def declare_variables(
self, ocp: OptimalControlProgram, nlp: NonLinearProgram, numerical_data_timeseries: dict[str, np.ndarray] = None
):
name = "Cn"
Expand Down Expand Up @@ -289,7 +289,7 @@ def prepare_ocp(
dynamics = DynamicsList()
for i in range(n_stim):
dynamics.add(
models[i].declare_ding_variables,
models[i].declare_variables,
dynamic_function=models[i].dynamics,
expand_dynamics=True,
expand_continuity=False,
Expand Down Expand Up @@ -650,7 +650,7 @@ def prepare_ocp(


@pytest.mark.parametrize("test_index", [0, 1, 2, 3])
def test_time_dependent_ding(test_index):
def test_time_dependent(test_index):
time_as_states = problem_dict[str(test_index)]["time_as_states"]
use_sx = problem_dict[str(test_index)]["use_sx"]

Expand Down

0 comments on commit d4e1831

Please sign in to comment.