Actual test files

dmwm · Mar 29, 2024 · 1c6b822 · 1c6b822
1 parent 7d0d928
commit 1c6b822
Show file tree

Hide file tree

Showing 3 changed files with 5 additions and 2 deletions.
diff --git a/src/python/WMComponent/RetryManager/Modifier/BaseModifier.py b/src/python/WMComponent/RetryManager/Modifier/BaseModifier.py
@@ -22,7 +22,7 @@ class BaseModifier(object):
     def __init__(self, config):
         object.__init__(self)
         self.backupPath = "./oldSandboxes"
-        self.tempFolder 
+        #self.tempFolder --> not being used? Initialized as what?
         self.sandboxPath = None
         self.config = config
 

diff --git a/src/python/WMComponent/RetryManager/Modifier/MemoryModifier.py b/src/python/WMComponent/RetryManager/Modifier/MemoryModifier.py
@@ -31,7 +31,7 @@ def changeSandbox(self, jobPKL, newMemory):
         for task in workHelper.getAllTasks():
             task.setMaxPSS(newMemory)
 
-        self.setWorkload(workload)
+        self.setWorkload(workload, jobPKL)
 
         return
 
@@ -40,6 +40,7 @@ def changeJobPkl(self, pklFile, jobPKL, newMemory):
         Modifies the pklFile job.pkl by changing the estimatedMemoryUsage to a new_memory value
 
         """
+        logging.info('MemoryModifier.changeJobPkl: Modifying {} job memory. Previous value: {}. New value: {}'.format(jobPKL['jobType'], jobPKL['estimatedMemoryUsage'], newMemory))
         jobPKL['estimatedMemoryUsage'] = newMemory 
         self.savePKL(pklFile, jobPKL)
 
@@ -98,4 +99,5 @@ def modifyJob(self, job):
             return
 
         else:
+            logging.info('MemoryModifier.changeMemory called successfully')
             self.changeMemory(job, settings)
diff --git a/src/python/WMComponent/RetryManager/RetryManagerPoller.py b/src/python/WMComponent/RetryManager/RetryManagerPoller.py
@@ -227,6 +227,7 @@ def processRetries(self, jobs, cooloffType):
 
         # Now we should have the jobs
         propList = self.selectRetryAlgo(jobList, cooloffType)
+        logging.info('Now calling selectJobModifier')
         propList = self.selectJobModifier(propList) #this propList has the modified jobs