Fixed formatting

HumanCompatibleAI · Apr 21, 2024 · 89aa083 · 89aa083
1 parent 0c9cdb4
commit 89aa083
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/src/overcooked_ai_py/mdp/overcooked_env.py b/src/overcooked_ai_py/mdp/overcooked_env.py
@@ -235,7 +235,7 @@ def print_state_transition(
         if fname is None:
             print(output_string)
         else:
-            with open(fname, "a", encoding='utf-8') as f:
+            with open(fname, "a", encoding="utf-8") as f:
                 print(output_string, file=f)
 
     ###################
@@ -445,7 +445,7 @@ def run_agents(
 
         if dir != None:
             fname = dir + "/roll_out_" + str(time.time()) + ".txt"
-            with open(fname, "w+", encoding='utf-8') as f:
+            with open(fname, "w+", encoding="utf-8") as f:
                 print(self, file=f)
         while not done:
             s_t = self.state

diff --git a/testing/overcooked_test.py b/testing/overcooked_test.py
@@ -1323,7 +1323,7 @@ def tearDown(self):
         shutil.rmtree(self.dummy_dir)
 
     def _assert_files_equal(self, file_1, file_2):
-        with open(file_1, "r", encoding='utf-8') as f:
+        with open(file_1, "r", encoding="utf-8") as f:
             lines_1 = f.readlines()
         with open(file_2, "r", encoding="utf-8") as f:
             lines_2 = f.readlines()