Dockerfile to get things running and tests passing

Dockerfile

@@ -0,0 +1,18 @@
+FROM python:3.8-slim-bullseye
+
+LABEL maintainer "DataMade <info@datamade.us>"
+
+RUN apt update && apt install -y gcc g++
+
+RUN mkdir /app
+WORKDIR /app
+
+# Copy the contents of the current host directory (i.e., our app code) into
+# the container.
+COPY . /app
+
+RUN pip install -e .
+
+RUN pip install -r requirements-dev.txt
+RUN pip install -r requirements-examples.txt
+

docker-compose.yml

@@ -0,0 +1,9 @@
+services:
+  app:
+    platform: linux/amd64
+    image: entity-embed
+    container_name: entity-embed
+    build: .
+    ports:
+      - 8888:8888
+    command: jupyter notebook --allow-root --ip=0.0.0.0 --NotebookApp.token='' --NotebookApp.password=''

requirements.txt

@@ -8,5 +8,6 @@ pytorch-metric-learning>=0.9.98,<1.0
 regex>=2020.11.13
 torch>=1.7.1,<1.9
 torchtext>=0.8,<0.10
-torchvision>=0.8.2<0.10
+torchvision>=0.8.2,<0.10
 tqdm>=4.53.0
+six