Я считаю, что нашел решение. Все сводится к рабочему каталогу, который не был правильно установлен :(
from google.colab import drive
drive.mount('/content/drive')
%cd /content/drive/My Drive/Colab Notebooks
Затем вы можете написать файл, например:
%%writefile calc.py
PI = 3.14
def add(a,b):
return (a + b)
print (add(2,3))
def area(radius):
return PI * radius * radius
print(area(5))
И, наконец, он может импортироваться:
import ctes3t