runtime error

ine 3, in <module> from tf_keras import __internal__ File "/usr/local/lib/python3.10/site-packages/tf_keras/__internal__/__init__.py", line 3, in <module> from tf_keras.__internal__ import backend File "/usr/local/lib/python3.10/site-packages/tf_keras/__internal__/backend/__init__.py", line 3, in <module> from tf_keras.src.backend import _initialize_variables as initialize_variables File "/usr/local/lib/python3.10/site-packages/tf_keras/src/__init__.py", line 21, in <module> from tf_keras.src import applications File "/usr/local/lib/python3.10/site-packages/tf_keras/src/applications/__init__.py", line 18, in <module> from tf_keras.src.applications.convnext import ConvNeXtBase File "/usr/local/lib/python3.10/site-packages/tf_keras/src/applications/convnext.py", line 33, in <module> from tf_keras.src.engine import sequential File "/usr/local/lib/python3.10/site-packages/tf_keras/src/engine/sequential.py", line 24, in <module> from tf_keras.src.engine import functional File "/usr/local/lib/python3.10/site-packages/tf_keras/src/engine/functional.py", line 33, in <module> from tf_keras.src.engine import training as training_lib File "/usr/local/lib/python3.10/site-packages/tf_keras/src/engine/training.py", line 48, in <module> from tf_keras.src.saving import saving_api File "/usr/local/lib/python3.10/site-packages/tf_keras/src/saving/saving_api.py", line 25, in <module> from tf_keras.src.saving.legacy import save as legacy_sm_saving_lib File "/usr/local/lib/python3.10/site-packages/tf_keras/src/saving/legacy/save.py", line 27, in <module> from tf_keras.src.saving.legacy.saved_model import load_context File "/usr/local/lib/python3.10/site-packages/tf_keras/src/saving/legacy/saved_model/load_context.py", line 68, in <module> tf.__internal__.register_load_context_function(in_load_context) AttributeError: module 'tensorflow._api.v2.compat.v2.__internal__' has no attribute 'register_load_context_function'. Did you mean: 'register_call_context_function'?

Container logs: