From bfb45f2cbd2a2a4581057e8e4079aa01c1a7ecdf Mon Sep 17 00:00:00 2001
From: uta <122957026+uta0x89@users.noreply.github.com>
Date: Thu, 6 Jul 2023 08:37:01 +0900
Subject: [PATCH] Fix errors in Mandatory Tasks of Benchmarks (#4893)

Co-authored-by: merwanehamadi <merwanehamadi@gmail.com>
---
 tests/challenges/basic_abilities/test_write_file.py | 2 ++
 tests/challenges/memory/test_memory_challenge_a.py  | 2 ++
 2 files changed, 4 insertions(+)

diff --git a/tests/challenges/basic_abilities/test_write_file.py b/tests/challenges/basic_abilities/test_write_file.py
index be1f0a93..2a202ee3 100644
--- a/tests/challenges/basic_abilities/test_write_file.py
+++ b/tests/challenges/basic_abilities/test_write_file.py
@@ -1,5 +1,6 @@
 import pytest
 
+from autogpt.config import Config
 from autogpt.workspace import Workspace
 from tests.challenges.challenge_decorator.challenge_decorator import challenge
 from tests.challenges.utils import get_workspace_path, run_challenge
@@ -17,6 +18,7 @@ USER_INPUTS = [
 
 @challenge()
 def test_write_file(
+    config: Config,
     patched_api_requestor: None,
     monkeypatch: pytest.MonkeyPatch,
     level_to_run: int,
diff --git a/tests/challenges/memory/test_memory_challenge_a.py b/tests/challenges/memory/test_memory_challenge_a.py
index 1130079b..bbd221f4 100644
--- a/tests/challenges/memory/test_memory_challenge_a.py
+++ b/tests/challenges/memory/test_memory_challenge_a.py
@@ -1,6 +1,7 @@
 import pytest
 from pytest_mock import MockerFixture
 
+from autogpt.config import Config
 from autogpt.workspace import Workspace
 from tests.challenges.challenge_decorator.challenge_decorator import challenge
 from tests.challenges.utils import get_workspace_path, run_challenge
@@ -12,6 +13,7 @@ USER_INPUT = "Use the command read_file to read the instructions_1.txt file\nFol
 
 @challenge()
 def test_memory_challenge_a(
+    config: Config,
     patched_api_requestor: MockerFixture,
     monkeypatch: pytest.MonkeyPatch,
     level_to_run: int,