Skip to content

Commit 1e62f15

Browse files
committed
Merge remote-tracking branch 'refs/remotes/origin/python_upgrades' into python_upgrades
2 parents d9f13ab + b65157e commit 1e62f15

File tree

1 file changed

+3
-3
lines changed
  • algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax

1 file changed

+3
-3
lines changed

algorithmic_efficiency/workloads/imagenet_resnet/imagenet_jax/randaugment.py

Lines changed: 3 additions & 3 deletions
Original file line numberDiff line numberDiff 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)

0 commit comments

Comments
 (0)