Skip to content

Commit

Permalink
Merge pull request DifferentiableUniverseInitiative#64 from aboucaud/…
Browse files Browse the repository at this point in the history
…remove-python2-future-imports

Remove __future__ imports
  • Loading branch information
EiffL authored Oct 17, 2020
2 parents e0c7b37 + ec580b0 commit a3ad52e
Show file tree
Hide file tree
Showing 14 changed files with 0 additions and 57 deletions.
4 changes: 0 additions & 4 deletions jax_cosmo/__init__.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# Cosmology in JAX
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from pkg_resources import DistributionNotFound
from pkg_resources import get_distribution

Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/angular_cl.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module contains functions to compute angular cls for various tracers
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from functools import partial

import jax.numpy as np
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/background.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module implements various functions for the background COSMOLOGY
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np

import jax_cosmo.constants as const
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/bias.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module contains implementations of galaxy bias
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np
from jax.tree_util import register_pytree_node_class

Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/core.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,3 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np
from jax.experimental.ode import odeint
from jax.tree_util import register_pytree_node_class
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/jax_utils.py
Original file line number Diff line number Diff line change
@@ -1,7 +1,3 @@
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np


Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/likelihood.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module implements a few likelihoods useful
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np
import jax.scipy as sp

Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/parameters.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module defines a few default cosmologies
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from functools import partial

from jax_cosmo.core import Cosmology
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/power.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module computes power spectra
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax
import jax.numpy as np

Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/probes.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module defines kernel functions for various tracers
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np
from jax import jit
from jax import vmap
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/redshift.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# Module to define redshift distributions we can differentiate through
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

from abc import ABC
from abc import abstractmethod

Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/sparse.py
Original file line number Diff line number Diff line change
Expand Up @@ -27,10 +27,6 @@
- :fun:`sparse_dot_sparse`
- :fun:`dense_dot_sparse_dot_dense`
"""
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import functools

import jax.numpy as np
Expand Down
4 changes: 0 additions & 4 deletions jax_cosmo/transfer.py
Original file line number Diff line number Diff line change
@@ -1,8 +1,4 @@
# This module contains various transfer functions from the literatu
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function

import jax.numpy as np

import jax_cosmo.background as bkgrd
Expand Down
5 changes: 0 additions & 5 deletions jax_cosmo/utils.py
Original file line number Diff line number Diff line change
@@ -1,9 +1,4 @@
# This defines a few utility functions
from __future__ import absolute_import
from __future__ import division
from __future__ import print_function


def z2a(z):
""" converts from redshift to scale factor """
return 1.0 / (1.0 + z)
Expand Down

0 comments on commit a3ad52e

Please sign in to comment.