Skip to content

Commit

Permalink
Fix formatting issue
Browse files Browse the repository at this point in the history
Signed-off-by: Chaurasiya, Payal <[email protected]>
  • Loading branch information
payalcha committed Nov 19, 2024
1 parent 98b8448 commit 411f5d0
Showing 1 changed file with 0 additions and 13 deletions.
13 changes: 0 additions & 13 deletions openfl/component/collaborator/collaborator.py
Original file line number Diff line number Diff line change
Expand Up @@ -177,19 +177,6 @@ def run(self):
self.tensor_db.clean_up(self.db_store_rounds)
if self.log_memory_usage:
# This is the place to check the memory usage of the collaborator
self.logger.info(
"*****************COLLABORATOR LOGS*******************************"
)
process = psutil.Process()
self.logger.info(process)
process_mem = round(process.memory_info().rss / (1024**2), 2)
self.logger.info("Collaborator Round: %s", round_number)
self.logger.info("Collaborator Process Mem: %s", process_mem)
self.logger.info(
"******************************************************************"
)

# NAD:This prints the data correctly : Get the Mem usage info here
memory_detail = self.get_memory_usage(
round_number, metric_origin=self.collaborator_name
)
Expand Down

0 comments on commit 411f5d0

Please sign in to comment.