From 7f2f11c765417216b44a19cdb9bb117fa9d23b5a Mon Sep 17 00:00:00 2001
From: Joshua David <jshuadvd@icloud.com>
Date: Thu, 4 Jul 2024 22:59:22 -0700
Subject: [PATCH] Import evaluate_passkey_retrieval for use in the train
 methond

---
 train.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/train.py b/train.py
index 5723aee..5eb74e4 100644
--- a/train.py
+++ b/train.py
@@ -17,6 +17,8 @@
 import wandb
 import os
 
+from evaluation import evaluate_passkey_retrieval
+
 reload(src.main)
 
 # Initialize the accelerator