From b740b428e5abaf4ca0bb1dc44c90541349bd8ea0 Mon Sep 17 00:00:00 2001
From: Alain Vaucher <ava@zurich.ibm.com>
Date: Fri, 5 Aug 2022 08:07:24 +0200
Subject: [PATCH] Fix incorrect import in training script

---
 training_scripts/launch_suzuki_miyaura_training.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/training_scripts/launch_suzuki_miyaura_training.py b/training_scripts/launch_suzuki_miyaura_training.py
index 87d7df0..e4e9eba 100644
--- a/training_scripts/launch_suzuki_miyaura_training.py
+++ b/training_scripts/launch_suzuki_miyaura_training.py
@@ -10,7 +10,8 @@
 from tqdm import tqdm
 from typing import List
 
-from rxn_yields.core import SmilesTokenizer, SmilesClassificationModel
+from rxn_yields.core import SmilesClassificationModel
+from rxnfp.tokenization import SmilesTokenizer
 import sklearn
 try:
     import wandb