Del (object) from 10 inc pytorch/inplace_abn/scripts/imagenet/utils.py

r-barnes · facebook-github-bot · commit 95df0904ae5d · 2023-09-05T17:50:44.000-07:00
Summary: Python3 makes the use of `(object)` in class inheritance unnecessary. Let's modernize our code by eliminating this.

Reviewed By: meyering

Differential Revision: D48957980

fbshipit-source-id: ef91fea76e0d4647d1c447429b8c377e81e121ab
diff --git a/opacus/utils/batch_memory_manager.py b/opacus/utils/batch_memory_manager.py
@@ -116,7 +116,7 @@ def wrap_data_loader(
     )
 
 
-class BatchMemoryManager(object):
+class BatchMemoryManager:
     """
     Context manager to manage memory consumption during training.
 

Original file line number	Diff line number	Diff line change
`@@ -116,7 +116,7 @@ def wrap_data_loader(`
`116`	`116`	`)`
`117`	`117`
`118`	`118`
`119`		`-class BatchMemoryManager(object):`
	`119`	`+class BatchMemoryManager:`
`120`	`120`	`"""`
`121`	`121`	`Context manager to manage memory consumption during training.`
`122`	`122`