Python import functions from module twice with different internal imports

Question

I have a module lib that needs numpy. So, for instance, let\'s say I have a hypothetical function that looks like

import numpy
def doS         


        

Answer 1

If your desire is to avoid duplicating your code base, make your interface a class instead. For example:

class using_numpy:
    import numpy

    @classmethod
    def do_something(cls, x):
        return cls.numpy.sqrt(x)

class using_autograd(using_numpy):
    from autograd import numpy

Now using_numpy.do_something will use numpy, and using_autograd.do_something will use autograd.numpy.

Alternatively, if you are uncomfortable with classmethod, you could make your interfaces instances of a class, for example:

class interface:
    def __init__(self, mdl):
        self.mdl = mdl

    def do_something(self, x):
        return self.mdl.sqrt(x)

import numpy
import autograd

with_numpy = interface(numpy)
with_autograd = interface(autograd.numpy)