From ed22d53895a9a737522d679d7656514430ede1b6 Mon Sep 17 00:00:00 2001 From: Madison May Date: Thu, 18 Dec 2014 00:56:12 -0500 Subject: [PATCH] Batch update --- indicoio/__init__.py | 7 ++++--- tests/local/test_local.py | 1 + 2 files changed, 5 insertions(+), 3 deletions(-) diff --git a/indicoio/__init__.py b/indicoio/__init__.py index 699ac87..6527b6e 100644 --- a/indicoio/__init__.py +++ b/indicoio/__init__.py @@ -16,10 +16,11 @@ from indicoio.images.features import image_features apis = ['political', 'posneg', 'sentiment', 'language', 'fer', 'facial_features', 'image_features', 'text_tags'] apis = dict((api, globals().get(api)) for api in apis) -local = {} +class Namespace(object): pass +local = Namespace() for api in apis: globals()[api] = partial(apis[api], config.api_root) globals()['batch_' + api] = partial(apis[api], config.api_root, batch=True) - local[api] = partial(apis[api], config.local_api_root) - local[api] = partial(apis[api], config.local_api_root, batch=True) + setattr(local, api, partial(apis[api], config.local_api_root)) + setattr(local, 'batch_' + api, partial(apis[api], config.local_api_root, batch=True)) diff --git a/tests/local/test_local.py b/tests/local/test_local.py index 48cc0f5..2f66ce6 100644 --- a/tests/local/test_local.py +++ b/tests/local/test_local.py @@ -1,4 +1,5 @@ import unittest +import os import numpy as np