@MSI Can you modify this very minimal Colab gist for your use case as I cannot reproduce your error in this minimal context:

  
      

      colab.research.google.com
  

  
    

Google Colaboratory 1