From d1ca84a3cb54002075d0a3e4f7794ab808084096 Mon Sep 17 00:00:00 2001
From: rdguha <rdguha@lbl.gov>
Date: Mon, 22 Jan 2024 10:09:51 -0800
Subject: [PATCH] small changes to correct linting errors

---
 src/atomate2/qchem/flows/core.py | 7 ++++---
 tests/conftest.py                | 4 +---
 tests/qchem/jobs/test_core.py    | 9 +++------
 3 files changed, 8 insertions(+), 12 deletions(-)

diff --git a/src/atomate2/qchem/flows/core.py b/src/atomate2/qchem/flows/core.py
index af10e8af9..a91b2ecbf 100644
--- a/src/atomate2/qchem/flows/core.py
+++ b/src/atomate2/qchem/flows/core.py
@@ -9,7 +9,6 @@
 
 from jobflow import Flow, Maker
 
-from atomate2.qchem.jobs.base import BaseQCMaker
 from atomate2.qchem.jobs.core import FreqMaker, OptMaker, TransitionStateMaker
 
 if TYPE_CHECKING:
@@ -17,6 +16,8 @@
 
     from pymatgen.core.structure import Molecule
 
+    from atomate2.qchem.jobs.base import BaseQCMaker
+
 # from atomate2.qchem.schemas.calculation import VaspObject
 # from atomate2.vasp.sets.core import HSEBSSetGenerator, NonSCFSetGenerator
 
@@ -95,7 +96,7 @@ def make(
 
     @classmethod
     def from_freq_and_opt_maker(
-        cls, freq_maker: BaseQCMaker, opt_maker=BaseQCMaker
+        cls, freq_maker: BaseQCMaker, opt_maker: BaseQCMaker
     ) -> Maker:
         """
         Instantiate the FrequencyFlatteningOptimizeMaker with a Freq and an Opt maker.
@@ -179,7 +180,7 @@ def make(
 
     @classmethod
     def from_freq_and_ts_maker(
-        cls, freq_maker: BaseQCMaker, ts_maker=BaseQCMaker
+        cls, freq_maker: BaseQCMaker, ts_maker: BaseQCMaker
     ) -> Maker:
         """
         Instantiate the FrequencyFlatteningOptimizeMaker with a Freq and a TS maker.
diff --git a/tests/conftest.py b/tests/conftest.py
index 402947994..4280e9073 100644
--- a/tests/conftest.py
+++ b/tests/conftest.py
@@ -16,8 +16,6 @@
 
 from atomate2.utils.log import initialize_logger
 
-# print(sys.executable)
-
 
 @pytest.fixture(scope="session")
 def test_dir():
@@ -46,7 +44,7 @@ def clean_dir(debug_mode):
     os.chdir(new_path)
     yield
     if debug_mode:
-        print(f"Tests ran in {new_path}")
+        print(f"Tests ran in {new_path}")  # noqa: T201
     else:
         os.chdir(old_cwd)
         shutil.rmtree(new_path)
diff --git a/tests/qchem/jobs/test_core.py b/tests/qchem/jobs/test_core.py
index 9357381f7..6e2d56f68 100644
--- a/tests/qchem/jobs/test_core.py
+++ b/tests/qchem/jobs/test_core.py
@@ -15,14 +15,11 @@
 
 from atomate2.qchem.jobs.core import FreqMaker, OptMaker, SinglePointMaker
 
-# curr_dir = Path(os.path.dirname(sys.argv[0]))
-
-# Specify the file name
-file_name = "H2O.xyz"
+# Find parent directory
+current_directory = Path(__file__).resolve().parent
+file_name = current_directory / "H2O.xyz"
 
 # Construct the full path
-mol_path = Path("tests/qchem/jobs/H2O.xyz")
-# H2O_structure = Molecule.from_file(mol_path)
 H2O_structure = Molecule.from_file(file_name)