build: fix typing/mypy issues
This commit is contained in:
		
							
								
								
									
										2
									
								
								tensorflow-stubs/keras/dtensor/experimental/__init__.pyi
									
									
									
									
									
										Normal file
									
								
							
							
						
						
									
										2
									
								
								tensorflow-stubs/keras/dtensor/experimental/__init__.pyi
									
									
									
									
									
										Normal 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
 | 
			
		||||
@@ -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
 | 
			
		||||
		Reference in New Issue
	
	Block a user