def _gen_dispatch(self, task): # Get dispatch based on type dispatch_type = self._get_config('mri-client', 'dispatch').lower() if dispatch_type == 'matplotlib-dispatch': folder = os.path.join( self._get_config('matplotlib-dispatch', 'save_img_folder')) dispatch = MatplotlibDispatch(task, folder) windows = self._get_config('matplotlib-dispatch', 'show_windows') if windows: show_windows = windows.lower() == 'true' else: show_windows = False dispatch.setup_display('iteration', ['iteration', 'loss', 'accuracy'], show_windows=show_windows) elif dispatch_type == 'mri-server-dispatch': url = self._get_config('mri-server-dispatch', 'url') username = self._get_config('mri-server-dispatch', 'username') password = self._get_config('mri-server-dispatch', 'password') dispatch = MriServerDispatch(task, url, username, password) dispatch.setup_display('iteration', ['iteration', 'loss', 'accuracy']) else: logging.error( 'Invalid configuration file, please select a dispatch') raise Exception( 'Invalid configuration file, please select a dispatch') return dispatch
def test_new_dispatch(self): server = MriServer("http://www.httpbin.com", "testuser", "testpass") task = {"title": "TEST", "id": "000112233"} dispatch = server.new_dispatch(task) test_against = MriServerDispatch(task, "http://www.httpbin.com", "testuser", "testpass") self.assertEqual(dispatch, test_against)