diff --git a/scripts/component_interface_value_fixtures.py b/scripts/component_interface_value_fixtures.py index 64f58cb9c..c421f59a4 100644 --- a/scripts/component_interface_value_fixtures.py +++ b/scripts/component_interface_value_fixtures.py @@ -87,8 +87,8 @@ def _create_civ_rich_algorithm_job(creator, interfaces): algorithm_image=ai, status=Evaluation.SUCCESS, time_limit=ai.algorithm.time_limit, - requires_gpu_type=ai.requires_gpu_type, - requires_memory_gb=ai.requires_memory_gb, + requires_gpu_type=ai.algorithm.requires_gpu_type, + requires_memory_gb=ai.algorithm.requires_memory_gb, ) chart_spec = { diff --git a/scripts/cost_fixtures.py b/scripts/cost_fixtures.py index 88e3263b3..1cf158e8f 100644 --- a/scripts/cost_fixtures.py +++ b/scripts/cost_fixtures.py @@ -180,8 +180,8 @@ def _create_submission(algorithm, challenge, archive_items): completed_at=now(), status=Job.SUCCESS, time_limit=ai.algorithm.time_limit, - requires_gpu_type=ai.requires_gpu_type, - requires_memory_gb=ai.requires_memory_gb, + requires_gpu_type=ai.algorithm.requires_gpu_type, + requires_memory_gb=ai.algorithm.requires_memory_gb, ) civ = ComponentInterfaceValue.objects.create( interface=ComponentInterface.objects.get(slug="results-json-file"), diff --git a/scripts/development_fixtures.py b/scripts/development_fixtures.py index c25d2ccba..23dd4978f 100644 --- a/scripts/development_fixtures.py +++ b/scripts/development_fixtures.py @@ -259,8 +259,8 @@ def _create_demo_challenge(users, algorithm): phase=phase, creator=users["demop"], algorithm_image=algorithm_image, - algorithm_requires_gpu_type=algorithm_image.requires_gpu_type, - algorithm_requires_memory_gb=algorithm_image.requires_memory_gb, + algorithm_requires_gpu_type=algorithm_image.algorithm.requires_gpu_type, + algorithm_requires_memory_gb=algorithm_image.algorithm.requires_memory_gb, ) submission.save() @@ -269,8 +269,8 @@ def _create_demo_challenge(users, algorithm): method=method, status=Evaluation.SUCCESS, time_limit=submission.phase.evaluation_time_limit, - requires_gpu_type=method.requires_gpu_type, - requires_memory_gb=method.requires_memory_gb, + requires_gpu_type=method.phase.requires_gpu_type, + requires_memory_gb=method.phase.requires_memory_gb, ) def create_result(evaluation, result: dict): @@ -383,8 +383,8 @@ def _create_algorithm_demo(users): algorithm_image=algorithm_image, status=Evaluation.SUCCESS, time_limit=60, - requires_gpu_type=algorithm_image.requires_gpu_type, - requires_memory_gb=algorithm_image.requires_memory_gb, + requires_gpu_type=algorithm_image.algorithm.requires_gpu_type, + requires_memory_gb=algorithm_image.algorithm.requires_memory_gb, ) algorithms_job.inputs.add(input_civ) diff --git a/scripts/pentest_fixtures.py b/scripts/pentest_fixtures.py index 391aeca80..e54fd0c01 100644 --- a/scripts/pentest_fixtures.py +++ b/scripts/pentest_fixtures.py @@ -475,8 +475,8 @@ def create_algorithm( ) ], time_limit=algorithm_image.algorithm.time_limit, - requires_gpu_type=algorithm_image.requires_gpu_type, - requires_memory_gb=algorithm_image.requires_memory_gb, + requires_gpu_type=algorithm_image.algorithm.requires_gpu_type, + requires_memory_gb=algorithm_image.algorithm.requires_memory_gb, ) on_commit(job.execute)