diff --git a/orchestration/flows/bl7012/ptycho_nersc.py b/orchestration/flows/bl7012/ptycho_nersc.py index c2aaf04..911231e 100644 --- a/orchestration/flows/bl7012/ptycho_nersc.py +++ b/orchestration/flows/bl7012/ptycho_nersc.py @@ -1,10 +1,3 @@ -import json -import logging -import time - -from authlib.integrations.requests_client import OAuth2Session -from authlib.oauth2.rfc7523 import PrivateKeyJWT - from orchestration.ptycho_jobscript import ( get_job_script, cdtool_args_string, diff --git a/orchestration/nersc.py b/orchestration/nersc.py index 27e1808..777b752 100644 --- a/orchestration/nersc.py +++ b/orchestration/nersc.py @@ -75,11 +75,6 @@ def init_directory_paths(self): self.home_path = f"/global/homes/{self.user().name[0]}/{self.user().name}" self.scratch_path = f"/pscratch/sd/{self.user().name[0]}/{self.user().name}" - def request_task_status(self): - """Could need session variable in class, if this function - is to be used, due to information access requirements.""" - pass - def request_job_status(self): self.job = self.perlmutter.job(jobid=self.jobid) @@ -106,9 +101,9 @@ def submit_job(self, job_script): self.has_ran = False self.job_script_string = job_script - #self.logger.info(f"Submitting job with script: {job_script}") + self.logger.info(f"Submitting job with script: {job_script}") self.job = self.perlmutter.submit_job(job_script) self.update_job_id() #self.update_job_state() - #self.logger.info(f"Submitted job id: {self.jobid}") + self.logger.info(f"Submitted job id: {self.jobid}")