Cannot install sonnet on mac m1

I install dm-sonnet and I see:
Requirement already satisfied: sonnet in
```
/Users/maryamkia/Homebrew/Caskroom/miniforge/base/envs/pytorch_x86/lib/python3.8/site-packages (0.1.6)
Requirement already satisfied: networkx==1.8.1 in /Users/maryamkia/Homebrew/Caskroom/miniforge/base/envs/pytorch_x86/lib/python3.8/site-packages (from sonnet) (1.8.1)
```

Note: you may need to restart the kernel to use updated packages.

after reseting kernel and running:

```
import sonnet as snt
print("Sonnet version {}".format(snt.__version__))

```
I got:

`AttributeError: module 'tensorflow' has no attribute 'GraphKeys'`
I work on mac M1