mirror of
https://github.com/gsi-upm/senpy
synced 2024-11-10 19:12:28 +00:00
21a5a3f201
* Fixed Options for extra_params in UI * Enhanced meta-programming for models * Plugins can be imported from a python file if they're named `senpy_<whatever>.py>` (no need for `.senpy` anymore!) * Add docstings and tests to most plugins * Read plugin description from the docstring * Refactor code to get rid of unnecessary `.senpy`s * Load models, plugins and utils into the main namespace (see __init__.py) * Enhanced plugin development/experience with utils (easy_test, easy_serve) * Fix bug in check_template that wouldn't check objects * Make model defaults a private variable * Add option to list loaded plugins in CLI * Update docs
38 lines
794 B
Python
38 lines
794 B
Python
from senpy import AnalysisPlugin
|
|
|
|
import multiprocessing
|
|
|
|
|
|
def _train(process_number):
|
|
return process_number
|
|
|
|
|
|
class Async(AnalysisPlugin):
|
|
'''An example of an asynchronous module'''
|
|
author = '@balkian'
|
|
version = '0.2'
|
|
async = True
|
|
|
|
def _do_async(self, num_processes):
|
|
pool = multiprocessing.Pool(processes=num_processes)
|
|
values = sorted(pool.map(_train, range(num_processes)))
|
|
|
|
return values
|
|
|
|
def activate(self):
|
|
self.value = self._do_async(4)
|
|
|
|
def analyse_entry(self, entry, params):
|
|
values = self._do_async(2)
|
|
entry.async_values = values
|
|
yield entry
|
|
|
|
test_cases = [
|
|
{
|
|
'input': 'any',
|
|
'expected': {
|
|
'async_values': [0, 1]
|
|
}
|
|
}
|
|
]
|