Fix scripts

comic · Oct 15, 2024 · 23abda6 · 23abda6
1 parent b884d5a
commit 23abda6
Show file tree

Hide file tree

Showing 4 changed files with 12 additions and 12 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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)