Installation Improvements

.dockerignore

@@ -0,0 +1,16 @@
+.git/
+
+build
+dist
+*.egg-info
+*.egg/
+*.pyc
+*.swp
+
+.tox
+.coverage
+html/*
+__pycache__
+
+# Compiled Documentation
+docs/_build

.travis.yml

@@ -1,16 +1,29 @@
 language: python
 
 python:
-  - 2.7
-  - 3.5
+  - 3.7
 
 sudo: required
 
 install:
-  - sudo apt-get install gfortran python-liblas libblas-dev liblapack-dev libatlas-dev
-  - pip install --upgrade pip
-  - pip install -r requirements.txt
-  - pip install .[tf]
+  - sudo apt-get update
+  # We do this conditionally because it saves us some downloading if the
+  # version is the same.
+  - if [[ "$TRAVIS_PYTHON_VERSION" == "2.7" ]]; then
+      wget https://repo.continuum.io/miniconda/Miniconda2-latest-Linux-x86_64.sh -O miniconda.sh;
+    else
+      wget https://repo.continuum.io/miniconda/Miniconda3-latest-Linux-x86_64.sh -O miniconda.sh;
+    fi
+  - bash miniconda.sh -b -p $HOME/miniconda
+  - source "$HOME/miniconda/etc/profile.d/conda.sh"
+  - hash -r
+  - conda config --set always_yes yes --set changeps1 no
+  - conda update -q conda
+  # Useful for debugging any issues with conda
+  - conda info -a
+
+  # Install deps 
+  - make install
 
 script:
   - make download-test-data

Dockerfile

@@ -0,0 +1,15 @@
+FROM continuumio/miniconda:4.7.12
+
+RUN apt-get update -y && \
+    apt-get install -y --no-install-recommends build-essential && \
+    apt-get clean && \
+    rm -rf /var/lib/apt/lists/*
+
+WORKDIR /yass
+
+ENV PATH /opt/conda/bin:$PATH
+
+COPY . /yass
+
+RUN make install
+

Makefile

@@ -1,6 +1,10 @@
 .DEFAULT_GOAL := help
 .PHONY: test integration-test
 
+install: ## Install Locally
+	conda env update -f environment.yml
+	cd src/gpu_bspline_interp && conda run -n yass python setup.py install --force
+	cd src/gpu_rowshift && conda run -n yass python setup.py install --force
 
 test: ## Run Tests
 	@echo "--> Running Tests"

environment.yml

@@ -0,0 +1,19 @@
+name: yass
+channels:
+  - conda-forge
+  - pytorch
+  - defaults
+dependencies:
+  - python=3.7
+  - conda-build
+  - cudatoolkit-dev=10.1
+  - pytorch==1.3.1
+  - numpydoc
+  - pytest-cov
+  - pytest-flake8
+  - pytest>=3.5.0
+  - sphinx>=1.6
+  - pip
+  - pip:
+    - sphinxcontrib-blockdiag
+    - -e .