I want to change the source code in a python package. Specifically, I would like to change the following line from the [source code](https://github.com/tensorflow/probability/blob/9e50fa39f37216dd084c24aa3dd14ab0d4dee926/tensorflow_probability/python/layers/conv_variational.py#L384:)

I want to edit:

```
outputs_shape = outputs.shape.as_list()

```

To:

```
outputs_shape = tf.shape(outputs)

```

I know how to do this in google Colab, but I am running my code on a remote server.

If I don't want to create a virtual environment and don't have access to the files from where it is downloaded, is there a way (**python code**) to modify an imported package if it is originally installed as `pip install tensorflow_probability`?