I'm getting errors using the Colab notebook example for generating the primitives. Here is the notebook: https://colab.research.google.com/drive/1vr6eZkHfIvHTNKwgStIOdIqK1Q3av28Y?usp=sharing