safetensors.tensorflow.save modifies its input argument

[allaisandrea]

System Info

python version: 3.10.13 (main, Nov 19 2023, 17:34:57) [GCC 9.3.0]
tensorflow version: 2.12.0
safetensors version: 0.4.3

Information

• The official example scripts
• My own modified scripts

Reproduction

import safetensors.tensorflow
import tensorflow as tf
tensors = {'tensor1': tf.constant([1, 2, 3])}
safetensors.tensorflow.save_file(tensors, '/tmp/tensors.safetensors')
assert isinstance(tensors['tensor1'], tf.Tensor), type(tensors['tensor1'])

The assertion triggers

Traceback (most recent call last):
  ...
    assert isinstance(tensors['tensor1'], tf.Tensor), type(tensors['tensor1'])
AssertionError: <class 'numpy.ndarray'>

This is because this function modifies its argument

Expected behavior

The assertion should not trigger.

[github-actions]

This issue is stale because it has been open 30 days with no activity. Remove stale label or comment or this will be closed in 5 days.