fix linter

ACEsuit · Aug 22, 2024 · c50de48 · c50de48
1 parent a7c98ad
commit c50de48
Show file tree

Hide file tree

Showing 2 changed files with 3 additions and 3 deletions.
diff --git a/mace/tools/multihead_tools.py b/mace/tools/multihead_tools.py
@@ -1,8 +1,8 @@
 import argparse
+import dataclasses
 import logging
 import os
 import urllib.request
-import dataclasses
 from typing import Any, Dict, List, Optional, Union
 
 import torch

diff --git a/tests/test_run_train.py b/tests/test_run_train.py
@@ -308,7 +308,7 @@ def test_run_train_multihead(tmp_path, fitting_configs):
         for sub_key, sub_value in value.items():
             yaml_str += f"    {sub_key}: {sub_value}\n"
     filename = tmp_path / "config.yaml"
-    with open(filename, "w") as file:
+    with open(filename, "w", encoding="utf-8") as file:
         file.write(yaml_str)
 
     mace_params = _mace_params.copy()
@@ -486,7 +486,7 @@ def test_run_train_foundation_multihead(tmp_path, fitting_configs):
         for sub_key, sub_value in value.items():
             yaml_str += f"    {sub_key}: {sub_value}\n"
     filename = tmp_path / "config.yaml"
-    with open(filename, "w") as file:
+    with open(filename, "w", encoding="utf-8") as file:
         file.write(yaml_str)
     mace_params = _mace_params.copy()
     mace_params["valid_fraction"] = 0.1