diff --git a/CHANGELOG.md b/CHANGELOG.md index 81b29546b8..2eb2f3a9ad 100644 --- a/CHANGELOG.md +++ b/CHANGELOG.md @@ -24,6 +24,7 @@ This project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.htm * GIB: Add missing import during group-in-a-box cudf layout of 0-degree nodes * Tests: SSO login tests catch more unexpected exns +* Embed: `predict_links` on cuda ## [0.28.6 - 2022-29-22] diff --git a/graphistry/embed_utils.py b/graphistry/embed_utils.py index 10798d70a3..ec83bd3913 100644 --- a/graphistry/embed_utils.py +++ b/graphistry/embed_utils.py @@ -329,7 +329,7 @@ def _score_triplets(self, triplets, threshold, anomalous, retain_old_edges, retu log(f"{triplets.shape[0]} triplets for inference") ############################################################ # the bees knees - scores = self._score(triplets) + scores = self._score(triplets).cpu() ############################################################ if len(triplets) > 1: if anomalous: diff --git a/mypy.ini b/mypy.ini index fd6b4d9ece..01cff103e8 100644 --- a/mypy.ini +++ b/mypy.ini @@ -1,5 +1,5 @@ [mypy] -python_version = 3.7 +python_version = 3.8 # TODO check tests exclude = graph_vector_pb2|versioneer|_version|graphistry/tests @@ -90,4 +90,4 @@ ignore_missing_imports = True ignore_missing_imports = True [mypy-cuml.*] -ignore_missing_imports = True \ No newline at end of file +ignore_missing_imports = True diff --git a/setup.py b/setup.py index 5817609f6a..d865612a62 100755 --- a/setup.py +++ b/setup.py @@ -33,7 +33,7 @@ def unique_flatten_dict(d): 'networkx': ['networkx>=2.5'], 'gremlin': ['gremlinpython'], 'bolt': ['neo4j', 'neotime'], - 'nodexl': ['openpyxl', 'xlrd'], + 'nodexl': ['openpyxl<3.1.1', 'xlrd'], 'jupyter': ['ipython'], }