from . import layers
from . import utils
from . import nputils
from . import tbutils
from . import distributions
from .utils import FileWriter
from .tbutils import function

import sys
from . import layers
from . import utils
from . import nputils
from . import tbutils
from . import distributions
from .utils import FileWriter
from .tbutils import function

if 'ipykernel' in sys.argv[0]:
    from . import nbutils

Add nbutils import to base import
Add nbutils import to base import

+ import sys
  from . import layers
  from . import utils
  from . import nputils
  from . import tbutils
  from . import distributions
  from .utils import FileWriter
  from .tbutils import function
  
+ if 'ipykernel' in sys.argv[0]:
+     from . import nbutils
+ 
Add nbutils import to base import
## Code Before:
from . import layers
from . import utils
from . import nputils
from . import tbutils
from . import distributions
from .utils import FileWriter
from .tbutils import function

## Instruction:
Add nbutils import to base import
## Code After:
import sys
from . import layers
from . import utils
from . import nputils
from . import tbutils
from . import distributions
from .utils import FileWriter
from .tbutils import function

if 'ipykernel' in sys.argv[0]:
    from . import nbutils
