Fix import

Author: thomaspinder

gpjax/__init__.py

@@ -1,7 +1,11 @@
 from jax.config import config
 config.update("jax_enable_x64", True)
-from .gps.priors import Prior
+from . import gps
+from .gps import Prior  
 from .kernel import RBF
 from .likelihoods import Gaussian
 from .mean_functions import ZeroMean
 from .utilities import save, load
+
+
+__version__ = "0.1.0"

setup.py

@@ -18,6 +18,11 @@ def parse_requirements_file(filename):
     license='LICENSE',
     description=
     'Building blocks of Gaussian processes. Made to enable rapid development for researchers.',
-    long_description=open('README.md').read(),
+    long_description="GPJax aims to provide a low-level interface to Gaussian process models. Code is written entirely in Jax and Objax to enhance readability, and structured so as to allow researchers to easily extend the code to suit their own needs. When defining GP prior in GPJax, the user need only specify a mean and kernel function. A GP posterior can then be realised by computing the product of our prior with a likelihood function. The idea behind this is that the code should be as close as possible to the maths that we would write on paper when working with GP models.",
     install_requires=parse_requirements_file("requirements.txt"),
-)
+    # entry_points={
+    #     "console_scripts": [
+    #         "gpjax=gpjax.__main__:main",
+    #     ]
+    # },
+)