Unverified Commit 92db4bd5 authored by Quan (Andy) Gan's avatar Quan (Andy) Gan Committed by GitHub
Browse files

fix (#3700)

parent da877df2
# CI docker for lint # CI docker for lint
# Adapted from github.com/dmlc/tvm/docker/Dockerfile.ci_lint # Adapted from github.com/dmlc/tvm/docker/Dockerfile.ci_lint
FROM ubuntu:16.04 FROM ubuntu:18.04
ENV DEBIAN_FRONTEND=noninteractive
COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh COPY install/ubuntu_install_core.sh /install/ubuntu_install_core.sh
RUN bash /install/ubuntu_install_core.sh RUN bash /install/ubuntu_install_core.sh
...@@ -11,4 +13,4 @@ RUN bash /install/ubuntu_install_python.sh ...@@ -11,4 +13,4 @@ RUN bash /install/ubuntu_install_python.sh
RUN apt-get install -y doxygen graphviz RUN apt-get install -y doxygen graphviz
RUN pip3 install cpplint==1.3.0 pylint==2.3.0 mypy RUN pip3 install cpplint==1.3.0 pylint==2.7.0 mypy
...@@ -661,12 +661,12 @@ class RangePartitionBook(GraphPartitionBook): ...@@ -661,12 +661,12 @@ class RangePartitionBook(GraphPartitionBook):
for ntype in ntypes: for ntype in ntypes:
ntype_id = ntypes[ntype] ntype_id = ntypes[ntype]
self._ntypes[ntype_id] = ntype self._ntypes[ntype_id] = ntype
assert all([ntype is not None for ntype in self._ntypes]), \ assert all(ntype is not None for ntype in self._ntypes), \
"The node types have invalid IDs." "The node types have invalid IDs."
for etype in etypes: for etype in etypes:
etype_id = etypes[etype] etype_id = etypes[etype]
self._etypes[etype_id] = etype self._etypes[etype_id] = etype
assert all([etype is not None for etype in self._etypes]), \ assert all(etype is not None for etype in self._etypes), \
"The edge types have invalid IDs." "The edge types have invalid IDs."
# This stores the node ID ranges for each node type in each partition. # This stores the node ID ranges for each node type in each partition.
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment