Skip to content

Commit c792c3b

Browse files
legacy optimizer 2
1 parent 7c9b7e1 commit c792c3b

File tree

1 file changed

+4
-1
lines changed

1 file changed

+4
-1
lines changed

tests/test_adda.py

Lines changed: 4 additions & 1 deletion
Original file line numberDiff line numberDiff line change
@@ -7,8 +7,11 @@
77
import tensorflow as tf
88
from tensorflow.keras import Sequential, Model
99
from tensorflow.keras.layers import Dense
10-
from tensorflow.keras.optimizers.legacy import Adam
1110
from tensorflow.keras.initializers import GlorotUniform
11+
try:
12+
from tensorflow.keras.optimizers.legacy import Adam
13+
except:
14+
from tensorflow.keras.optimizers import Adam
1215

1316
from adapt.feature_based import ADDA
1417

0 commit comments

Comments
 (0)