Update tensorflow_script_mode_california_housing_local_training_and_serving.py

Author: eitansela

tensorflow_script_mode_california_housing_local_training_and_serving/tensorflow_script_mode_california_housing_local_training_and_serving.py

@@ -86,8 +86,8 @@ def main():
  role=DUMMY_IAM_ROLE,
  instance_count=1,
  instance_type='local',
- framework_version='2.4.1',
- py_version='py37')
+ framework_version='2.8',
+ py_version='py39')
 
  inputs = {'train': 'file://./data/train', 'test': 'file://./data/test'}
  california_housing_estimator.fit(inputs)