Update pyproject.toml

datamol-io · Sep 4, 2023 · 8be0f9f · 8be0f9f
1 parent fa5becc
commit 8be0f9f
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/pyproject.toml b/pyproject.toml
@@ -64,7 +64,7 @@ dependencies = [
 
 [project.scripts]
 graphium = "graphium.cli.main:app"
-graphium-train = "graphium.cli.train_finetune:cli"
+graphium-train = "graphium.cli.train_finetune_test:cli"
 
 [project.urls]
 Website = "https://graphium.datamol.io/"