【備忘】Google Colaboratoryでカスタムモジュールをインポート

手順

1. Google Driveをマウント

from google.colab import drive
drive.mount('/content/drive')

2. sys.pathl.append()でパスを通してインポート

例として次のようなファイル構成だったとします。

• My Drive
  • colab Notebooks
    • module
      • utils.py
      • my_modules.py

※：「MY Drive」がGoogle Driveのルートディレクトリ

このファイル構成でカスタムモジュールをimportするには次のようにsys.path.append()でモジュールの親ディレクトリまでのパスをシステムに伝える必要があります。

import sys
ROOT_PATH = 'drive/My Drive/ colab Notebooks/module/'
sys.path.append(ROOT_PATH)

import utils
import my_modules

これでModuleNotFoundError: No module named 'utils'のようなエラーが出なければOK。

モジュールの変更を反映する

Google Drive を再度マウントする必要がある。 Google Colaboratory のメニュータブから「Factory reset runtime」を選択し、ランタイムをリセットする。 そしていつものようにマウントするためのコードを実行すれば良い。

from google.colab import drive
drive.mount('/content/drive')

認証コードを入力してマウントが完了すれば、変更が反映されているはずです。

以上