build: fix typing/mypy issues

This commit is contained in:
2022-10-27 14:25:09 +02:00
parent 47653d723c
commit d7756a7184
82 changed files with 6552 additions and 46 deletions

View File

@ -0,0 +1 @@
from keras.api._v2.keras.dtensor import experimental as experimental

View File

@ -0,0 +1,2 @@
from keras.api._v2.keras.dtensor.experimental import optimizers as optimizers
from keras.dtensor.layout_map import LayoutMap as LayoutMap, layout_map_scope as layout_map_scope

View File

@ -0,0 +1 @@
from keras.dtensor.optimizers import Adadelta as Adadelta, Adagrad as Adagrad, Adam as Adam, AdamW as AdamW, RMSprop as RMSprop, SGD as SGD