refactor: Use Path.cwd instead of os.path.curdir

src/caustics/sims/state_dict.py

@@ -2,7 +2,6 @@
 from collections import OrderedDict
 from typing import Any, Dict, Optional
 from pathlib import Path
-import os
 
 from torch import Tensor
 import torch
@@ -185,7 +184,7 @@ def save(self, file_path: Optional[str] = None) -> str:
             The final path of the saved file
         """
         if not file_path:
-            file_path = Path(os.path.curdir) / self.__st_file
+            file_path = Path.cwd() / self.__st_file
         elif isinstance(file_path, str):
             file_path = Path(file_path)
 

tests/sims/test_state_dict.py

@@ -1,6 +1,5 @@
 from pathlib import Path
 from tempfile import TemporaryDirectory
-import os
 import sys
 
 import pytest
@@ -138,7 +137,7 @@ def test_st_file_string(self, simple_state_dict):
 
     def test_save(self, simple_state_dict):
         # Check for default save path
-        expected_fpath = Path(os.path.curdir) / simple_state_dict._StateDict__st_file
+        expected_fpath = Path.cwd() / simple_state_dict._StateDict__st_file
         default_fpath = simple_state_dict.save()
 
         assert Path(default_fpath).exists()