diff --git a/.gitignore b/.gitignore index 3656e9c..6ee36b0 100644 --- a/.gitignore +++ b/.gitignore @@ -144,4 +144,6 @@ Temporary Items .apdisk node_modules/ -.localstack/ \ No newline at end of file +.localstack/ +dbdata/ +s3/ \ No newline at end of file diff --git a/docker-compose.yml b/docker-compose.yml index 199b0ca..b2ac356 100644 --- a/docker-compose.yml +++ b/docker-compose.yml @@ -1,15 +1,50 @@ version: '3.2' services: - localstack: + s3: image: minio/minio:latest container_name: aws-s3 ports: - 9000:9000 environment: - MINIO_ACCESS_KEY=${AWS_ACCESS_KEY_ID} - - MINIO_SECRET_KEY=${AWS_SECRET_KEY} + - MINIO_SECRET_KEY=${AWS_SECRET_ACCESS_KEY} command: server /date - #volumes: - # - './.localstack:/tmp/localstack' - # - '/var/run/docker.sock:/var/run/docker.sock' \ No newline at end of file + networks: + - A + volumes: + - ./s3:/date + db: + restart: always + image: mysql/mysql-server:5.7.28 + container_name: mlflow_db + expose: + - "3306" + environment: + - MYSQL_DATABASE=${MYSQL_DATABASE} + - MYSQL_USER=${MYSQL_USER} + - MYSQL_PASSWORD=${MYSQL_PASSWORD} + - MYSQL_ROOT_PASSWORD=${MYSQL_ROOT_PASSWORD} + volumes: + - ./dbdata:/var/lib/mysql + networks: + - A + mlflow: + container_name: tracker_mlflow + image: tracker_ml + build: + context: ./mlflow + dockerfile: Dockerfile + ports: + - "5000:5000" + environment: + - AWS_ACCESS_KEY_ID=${AWS_ACCESS_KEY_ID} + - AWS_SECRET_ACCESS_KEY=${AWS_SECRET_ACCESS_KEY} + - AWS_DEFAULT_REGION=${AWS_REGION} + networks: + - A + + entrypoint: mlflow server --backend-store-uri mysql+pymysql://${MYSQL_USER}:${MYSQL_PASSWORD}@db:3306/${MYSQL_DATABASE} --default-artifact-root http://localhost:9000/${AWS_BUCKET_NAME}/mlflow/ -h 0.0.0.0 +networks: + A: + driver: bridge \ No newline at end of file diff --git a/mlflow/Dockerfile b/mlflow/Dockerfile index e2ca90a..9da8b46 100644 --- a/mlflow/Dockerfile +++ b/mlflow/Dockerfile @@ -3,4 +3,4 @@ FROM continuumio/miniconda3:latest ADD . /app WORKDIR /app -RUN pip install mlflow \ No newline at end of file +RUN pip install mlflow boto3 pymysql \ No newline at end of file