Merge pull request #72 from NERSC/job-models-rebuild

Add model_rebuild for job models

src/sfapi_client/_async/client.py

@@ -512,3 +512,10 @@ def resources(self) -> AsyncResources:
 AsyncRemotePath.model_rebuild()
 AsyncRole.model_rebuild()
 AsyncGroupMember.model_rebuild()
+
+# Ensure that the job models are built, we need to import here to
+# avoid circular imports
+from .jobs import AsyncJobSacct, AsyncJobSqueue  # noqa: E402
+
+AsyncJobSqueue.model_rebuild()
+AsyncJobSacct.model_rebuild()

src/sfapi_client/_async/paths.py

@@ -165,7 +165,10 @@ async def download(self, binary=False) -> IO[AnyStr]:
 
     @staticmethod
     async def _ls(
-        compute: "Compute", path, directory=False, filter_dots=True  # noqa: F821
+        compute: "AsyncCompute",  # noqa: F821
+        path,
+        directory=False,
+        filter_dots=True,  # noqa: F821
     ) -> List["RemotePath"]:  # noqa: F821
         r = await compute.client.get(f"utilities/ls/{compute.name}/{path}")
 

src/sfapi_client/_async/users.py

@@ -31,7 +31,8 @@ class AsyncUser(UserBase):
     @staticmethod
     @check_auth
     async def _fetch_user(
-        client: "AsyncClient", username: Optional[str] = None  # noqa: F821
+        client: "AsyncClient",  # noqa: F821
+        username: Optional[str] = None,  # noqa: F821
     ):  # noqa: F821
         url = "account/"
         if username is not None:

src/sfapi_client/_sync/client.py

@@ -512,3 +512,9 @@ def resources(self) -> Resources:
 RemotePath.model_rebuild()
 Role.model_rebuild()
 GroupMember.model_rebuild()
+
+# Ensure that the job models are built, we need to import here to
+# avoid circular imports
+from .jobs import JobSacct, JobSqueue
+JobSqueue.model_rebuild()
+JobSacct.model_rebuild()