TFnode on TensorflowonSpark 2.2.5

[cmilanes93]

I'm running some code on Microsoft Fabric, and I use the following line:
def map_fun(tf_args, ctx):
cluster, server = TFNode.start_cluster_server(ctx)
print('ctx')
if ctx.job_name == "ps":
server.join()
else:
print("Hello from worker", ctx.task_index)

I'm getting an error regarding TFnode, does anyone know why this could be?
TensorFlow version:2.12
TensorflowonSpark:2.2.5