update import path in randaugment.py

mlcommons · Feb 11, 2025 · 4dadef0 · 4dadef0
1 parent 1f72cb3
commit 4dadef0
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/algoperf/workloads/imagenet_resnet/imagenet_jax/randaugment.py b/algoperf/workloads/imagenet_resnet/imagenet_jax/randaugment.py
@@ -7,11 +7,11 @@
 import inspect
 import math
 
-from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
+from algoperf.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     rotate_img
-from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
+from algoperf.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     transform
-from algorithmic_efficiency.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
+from algoperf.workloads.imagenet_resnet.imagenet_jax.custom_tf_addons import \
     translate
 import tensorflow as tf