Skip to content

Commit

Permalink
Removes tqdm from OrientationMaps for CI to work
Browse files Browse the repository at this point in the history
  • Loading branch information
hernanbari committed Oct 18, 2018
1 parent 27a1afe commit 8d65bd6
Showing 1 changed file with 1 addition and 3 deletions.
4 changes: 1 addition & 3 deletions pylissom/utils/orientation_maps.py
Original file line number Diff line number Diff line change
Expand Up @@ -9,7 +9,6 @@
import numpy as np
import torch
from skimage.transform import rotate
from tqdm import tqdm, tqdm_notebook

from pylissom.utils.stimuli import translate, generate_horizontal_bar
from pylissom.utils.training.pipeline import Pipeline
Expand All @@ -21,11 +20,10 @@ def __init__(self, model, inputs, use_tqdm_notebook=True):
self.use_tqdm_notebook = use_tqdm_notebook
self.model = model
self.inputs = inputs
self.tqdm = tqdm_notebook if self.use_tqdm_notebook else tqdm

def maximum_activations(self, model, inputs):
activations = []
for inp in self.tqdm(inputs):
for inp in inputs:
inp = Pipeline.process_input(inp)
act = model(inp)
activations.append(act)
Expand Down

0 comments on commit 8d65bd6

Please sign in to comment.