"ValueError: (2112) Data Validation Error when validating your Snowpark DataFrame." in Tasty Bytes - Snowpark 101 for Data Science

[Sygyzmundovych]

Describe the bug
ValueError: (2112) Data Validation Error when validating your Snowpark DataFrame. If using the normalized names from model signatures, there are the following errors: [ValueError('(2112) Data Validation Error in feature AVG_LOCATION_SHIFT_SALES: Feature type DataType.INT64 is not met by column AVG_LOCATION_SHIFT_SALES because of its original type DoubleType()')] If using the inferred names from model signatures, there are the following errors: [ValueError('(2112) Data Validation Error in feature AVG_LOCATION_SHIFT_SALES: Feature type DataType.INT64 is not met by column AVG_LOCATION_SHIFT_SALES because of its original type DoubleType()')]
Traceback:

File "Cell [make_predictions]", line 2, in <module>
    inference_df = reg.get_model("SHIFT_SALES_FORECASTER").default.run(X=location_predictions_df, function_name="predict")
File "/usr/lib/python_udf/7cbda7fbb70b4408a98b12f1759f1bf28c5046e1a1e8b0d5806445f16d75907a/lib/python3.9/site-packages/snowflake/ml/_internal/telemetry.py", line 529, in wrap
    raise e.original_exception from e

URL of where you see the bug
https://github.com/Snowflake-Labs/sfguide-tasty-bytes-snowpark-101-for-data-science/blob/main/tasty_bytes_snowpark_101.ipynb

To Reproduce
Run the notebook as described in https://quickstarts.snowflake.com/guide/tasty_bytes_snowpark_101_for_data_science/#2

Expected behavior
No run-time error

Screenshots

[Sygyzmundovych]

And then the same error impacts the Streamlite app:

@sfc-gh-chammond