[pre-commit.ci] auto fixes from pre-commit.com hooks

pre-commit-ci[bot] · pre-commit-ci[bot] · commit 5612001c3d58 · 2025-05-05T18:44:35.000Z
for more information, see https://pre-commit.ci
diff --git a/torchinfo/torchinfo.py b/torchinfo/torchinfo.py
@@ -514,7 +514,7 @@ def get_total_memory_used(data: CORRECTED_INPUT_DATA_TYPE) -> int:
             else sum
         ),
     )
-    return cast(int, result)
+    return cast("int", result)
 
 
 def get_input_tensor(

Original file line number	Diff line number	Diff line change
`@@ -514,7 +514,7 @@ def get_total_memory_used(data: CORRECTED_INPUT_DATA_TYPE) -> int:`
`514`	`514`	`else sum`
`515`	`515`	`),`
`516`	`516`	`)`
`517`		`- return cast(int, result)`
	`517`	`+ return cast("int", result)`
`518`	`518`
`519`	`519`
`520`	`520`	`def get_input_tensor(`