uoicorr_imports.py

# %load uoicorr_imports.py
%load_ext autoreload
%autoreload 2

import h5py

import numpy as np
import matplotlib.pyplot as plt

import sys, os, pdb

if os.name == 'posix':
    pass
else:

    # Hack to import pyuoi
    parent_path, current_dir = os.path.split(os.path.abspath('.'))
    while current_dir not in ['nse']:
        parent_path, current_dir = os.path.split(parent_path)
    p = os.path.join(parent_path, current_dir)
    # Add analysis
    if p not in sys.path:
        sys.path.append(p)

    import subprocess
    hname = subprocess.check_output('hostname')
    
    if 'ankitnse'.encode() in hname:

        if '%s/uoicorr' % p not in sys.path:
            sys.path.append('%s/uoicorr' % p)
        if '%s/PyUoI' % p not in sys.path:
            sys.path.append('%s/PyUoI' % p)

    elif os.name == 'posix':

        pass

    else:
        # And standard list of subdirectories
        if '%s\\pyuoi' % p not in sys.path:
            sys.path.append('%s\\pyuoi' % p)
        if '%s\\uoicorr' % p not in sys.path:
            sys.path.append('%s\\uoicorr' % p)

from postprocess import postprocess_file, postprocess_dir
from utils import FNR, FPR, selection_accuracy