From 3255e885304c2df12ce256de78875f6e5988dc79 Mon Sep 17 00:00:00 2001 From: Keenan Pepper Date: Tue, 29 Oct 2024 13:02:58 -0700 Subject: [PATCH] Add missing pathlib import --- codebook_features/train_codebook.py | 1 + 1 file changed, 1 insertion(+) diff --git a/codebook_features/train_codebook.py b/codebook_features/train_codebook.py index 451e425..fcb8902 100755 --- a/codebook_features/train_codebook.py +++ b/codebook_features/train_codebook.py @@ -11,6 +11,7 @@ import torch import transformers import wandb +import pathlib from codebook_features import models, run_clm from codebook_features import trainer as cb_trainer