X-Git-Url: http://git.ithinksw.org/philo.git/blobdiff_plain/c9295747198ef8b64612408539b48aca329cff79..b3e16b23802c9bc5a99193f9e9551b4378ee1119:/contrib/sobol/search.py diff --git a/contrib/sobol/search.py b/contrib/sobol/search.py index 36c2b5d..33603f3 100644 --- a/contrib/sobol/search.py +++ b/contrib/sobol/search.py @@ -42,8 +42,9 @@ class SearchRegistry(object): def register(self, search, slug=None): slug = slug or search.slug if slug in self._registry: - if self._registry[slug] != search: - raise RegistrationError("A different search is already registered as `%s`") + registered = self._registry[slug] + if registered.__module__ != search.__module__: + raise RegistrationError("A different search is already registered as `%s`" % slug) else: self._registry[slug] = search @@ -361,7 +362,7 @@ else: def parse_response(self, response, limit=None): strainer = self.strainer soup = BeautifulSoup(response, parseOnlyThese=strainer) - return self.parse_results(soup[:limit]) + return self.parse_results(soup.findAll(recursive=False, limit=limit)) def parse_results(self, results): """