From 75f5348a58fc3af5400211f19cb3f3162ab65b63 Mon Sep 17 00:00:00 2001
From: Guillaume Chervet <guillaume.chervet@gmail.com>
Date: Fri, 13 Sep 2024 14:43:25 +0200
Subject: [PATCH] refactor(docker): update docker

---
 poetry-install.sh               | 2 ++
 train/extraction/pyproject.toml | 2 +-
 train/test/pyproject.toml       | 2 +-
 3 files changed, 4 insertions(+), 2 deletions(-)

diff --git a/poetry-install.sh b/poetry-install.sh
index 57735926..8966a238 100644
--- a/poetry-install.sh
+++ b/poetry-install.sh
@@ -11,6 +11,8 @@ function add_dependency {
 function add_dependencies {
     add_dependency "packages/inference/" "production/api/packages"
     add_dependency "packages/extraction/" "production/api/packages"
+    add_dependency "packages/extraction/" "train/extraction/packages"
+    add_dependency "packages/inference/" "train/test/packages"
 }
 
 add_dependencies
diff --git a/train/extraction/pyproject.toml b/train/extraction/pyproject.toml
index 27f5173b..6ba108bc 100644
--- a/train/extraction/pyproject.toml
+++ b/train/extraction/pyproject.toml
@@ -7,7 +7,7 @@ authors = ["Guillaume Chervet <guillaume.chervet@gmail.com>"]
 [tool.poetry.dependencies]
 python = "~3.10"
 pymupdf = "1.23.3"
-mlopspython-extraction = {path = "./packages/mlopspython_extraction-0.0.0-py3-none-any.whl"}
+mlopspython-extraction = {path = "packages/extraction"}
 mldesigner = "0.1.0b15"
 mlflow = "2.7.1"
 azureml-mlflow = "1.53.0"
diff --git a/train/test/pyproject.toml b/train/test/pyproject.toml
index a502e87e..e2a078f8 100644
--- a/train/test/pyproject.toml
+++ b/train/test/pyproject.toml
@@ -15,7 +15,7 @@ mldesigner = "0.1.0b15"
 mlflow = "2.7.1"
 azureml-mlflow = "1.53.0"
 azure-ai-ml = "1.10.1"
-mlopspython-inference = {path = "./packages/mlopspython_inference-0.0.0-py3-none-any.whl"}
+mlopspython-inference = {path = "packages/inference"}
 
 [tool.poetry.group.dev.dependencies]
 coverage = "^7.3.1"