improve intellisense

bob/learn/tensorflow/scripts/fit.py

@@ -151,14 +151,14 @@ def fit(
         validation_data = None
 
         if strategy_fn is None:
-            model = model_fn()
+            model: tf.keras.Model = model_fn()
             x = train_input_fn()
             if eval_input_fn is not None:
                 validation_data = eval_input_fn()
         else:
             strategy = strategy_fn()
             with strategy.scope():
-                model = model_fn()
+                model: tf.keras.Model = model_fn()
                 x = strategy.distribute_datasets_from_function(train_input_fn)
                 if eval_input_fn is not None:
                     validation_data = strategy.distribute_datasets_from_function(