File tree Expand file tree Collapse file tree 1 file changed +3
-3
lines changed
algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax Expand file tree Collapse file tree 1 file changed +3
-3
lines changed Original file line number Diff line number Diff line change @@ -482,13 +482,13 @@ def _parse_policy_info(name,
482482
483483 # Check to see if prob is passed into function. This is used for operations
484484 # where we alter bboxes independently.
485- if 'prob' in inspect .getargspec (func )[0 ]:
485+ if 'prob' in inspect .getfullargspec (func )[0 ]:
486486 args = tuple ([prob ] + list (args ))
487487
488488 # Add in replace arg if it is required for the function that is being called.
489- if 'replace' in inspect .getargspec (func )[0 ]:
489+ if 'replace' in inspect .getfullargspec (func )[0 ]:
490490 # Make sure replace is the final argument
491- assert 'replace' == inspect .getargspec (func )[0 ][- 1 ]
491+ assert 'replace' == inspect .getfullargspec (func )[0 ][- 1 ]
492492 args = tuple (list (args ) + [replace_value ])
493493
494494 return (func , prob , args )
You can’t perform that action at this time.
0 commit comments