328d01bb53b37f293b7f4090293ea6395c55fc40,geomstats/backend/__init__.py,,,#,4

Before Change

    from .tensorflow import *  // NOQA
    from . import tensorflow_linalg as linalg  // NOQA
    from . import tensorflow_random as random  // NOQA
elif _BACKEND == "pytorch":
    raise NotImplementedError("pytorch backend not implemented yet")


def backend():
    return _BACKEND

After Change

    from . import numpy_random as random
    from . import numpy_testing as testing
elif _BACKEND == "pytorch":
    sys.stderr.write("Using pytorch backend\n")
    from .pytorch import *  // NOQA
    from . import pytorch_linalg as linalg  // NOQA
    from . import pytorch_random as random  // NOQA
elif _BACKEND == "tensorflow":

Instances