Developing on prefect issue with new version 3.4.9

[Cyber-Oscar]

Good morning!
I created a flow on prefect in Python using version 3.4.4. I have updated the prefect to 3.4.9, and I'm encountering an issue with TaskRunInput during deployment.

Encountered exception during execution: ImportError("cannot import name 'TaskRunInput' from 'prefect.client.schemas.objects' (/home/prefect/.local/lib/python3.11/site-packages/prefect/client/schemas/objects.py)")
Traceback (most recent call last):
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 765, in run_context
self.handle_timeout(exc)
^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1373, in run_flow_sync
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 785, in call_flow_fn
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/utilities/callables.py", line 210, in call_with_parameters
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/tmp/tmp92r3_1hiprefect/project/main.py", line 9, in initialize
download_flow.download_product_flow(numero)
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flows.py", line 1702, in call
it's rendered inline, otherwise in a new window as a PNG.
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1530, in run_flow
except:

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1375, in run_flow_sync
async def run_flow_async(

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 351, in result
return self._raised
^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 765, in run_context
self.handle_timeout(exc)
^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1373, in run_flow_sync
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 785, in call_flow_fn
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/utilities/callables.py", line 210, in call_with_parameters
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/tmp/tmp92r3_1hiprefect/project/flows/flow.py", line 66, in download_product_flow
product_data = mytask.download_product()
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/tasks.py", line 1137, in call
wait_for: Upstream task futures to wait for before starting the task
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/task_engine.py", line 40, in
from prefect.client.schemas.objects import State, TaskRunInput
ImportError: cannot import name 'TaskRunInput' from 'prefect.client.schemas.objects' (/home/prefect/.local/lib/python3.11/site-packages/prefect/client/schemas/objects.py)
10:21:51 AM
prefect.flow_runs
Finished in state Failed("Flow run encountered an exception: ImportError: cannot import name 'TaskRunInput' from 'prefect.client.schemas.objects' (/home/prefect/.local/lib/python3.11/site-packages/prefect/client/schemas/objects.py)")
10:21:51 AM
prefect.flow_runs
Engine execution exited with unexpected exception
Traceback (most recent call last):
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1530, in run_flow
except:

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1375, in run_flow_sync
async def run_flow_async(

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 351, in result
return self._raised
^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 765, in run_context
self.handle_timeout(exc)
^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1373, in run_flow_sync
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 785, in call_flow_fn
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/utilities/callables.py", line 210, in call_with_parameters
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/tmp/tmp92r3_1hiprefect/project/main.py", line 9, in initialize
download_flow.download_product_flow(numero)
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flows.py", line 1702, in call
it's rendered inline, otherwise in a new window as a PNG.
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1530, in run_flow
except:

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1375, in run_flow_sync
async def run_flow_async(

File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 351, in result
return self._raised
^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 765, in run_context
self.handle_timeout(exc)
^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 1373, in run_flow_sync
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/flow_engine.py", line 785, in call_flow_fn
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/utilities/callables.py", line 210, in call_with_parameters
return fn(*args, **kwargs)
^^^^^^^^^^^^^^^^^^^
File "/tmp/tmp92r3_1hiprefect/project/flows/flow.py", line 66, in download_product_flow
product_data = mytask.download_product()
^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/tasks.py", line 1137, in call
wait_for: Upstream task futures to wait for before starting the task
^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^^
File "/home/prefect/.local/lib/python3.11/site-packages/prefect/task_engine.py", line 40, in
from prefect.client.schemas.objects import State, TaskRunInput
ImportError: cannot import name 'TaskRunInput' from 'prefect.client.schemas.objects' (/home/prefect/.local/lib/python3.11/site-packages/prefect/client/schemas/objects.py)

I am using

from prefect import task

to run a task

[Cyber-Oscar]

#18394

Upgrade the prefect in both client and worker to fix the problem