[652] Fix types.

cqfn · Nov 20, 2020 · af9d2c6 · af9d2c6
1 parent 887cec7
commit af9d2c6
Show file tree

Hide file tree

Showing 6 changed files with 14 additions and 12 deletions.
diff --git a/aibolit/ast_framework/block_statement_graph/_block_extractors.py b/aibolit/ast_framework/block_statement_graph/_block_extractors.py
@@ -1,6 +1,7 @@
 from typing import Callable, Dict, List, Optional, NamedTuple, Union
 
-from veniq.ast_framework import ASTNode, ASTNodeType
+from ..ast_node import ASTNode
+from ..ast_node_type import ASTNodeType
 from .constants import BlockReason
 
 

diff --git a/aibolit/ast_framework/block_statement_graph/_nodes_factory.py b/aibolit/ast_framework/block_statement_graph/_nodes_factory.py
@@ -1,5 +1,5 @@
 from typing import Callable, Union
-from networkx import DiGraph, dfs_labeled_edges
+from networkx import DiGraph, dfs_labeled_edges  # type: ignore
 
 from .statement import Statement
 from .block import Block

diff --git a/aibolit/ast_framework/block_statement_graph/block.py b/aibolit/ast_framework/block_statement_graph/block.py
@@ -1,7 +1,7 @@
-from networkx import DiGraph
+from networkx import DiGraph  # type: ignore
 from typing import Any, Callable, Iterator, Optional, TYPE_CHECKING
 
-from veniq.ast_framework import ASTNode
+from ..ast_node import ASTNode
 from .constants import BLOCK_REASON, ORIGIN_STATEMENT, NODE, BlockReason, NodeId
 
 if TYPE_CHECKING:

diff --git a/aibolit/ast_framework/block_statement_graph/builder.py b/aibolit/ast_framework/block_statement_graph/builder.py
@@ -1,7 +1,8 @@
 from typing import Dict, TYPE_CHECKING
-from networkx import DiGraph
+from networkx import DiGraph  # type: ignore
 
-from veniq.ast_framework import AST, ASTNode
+from ..ast import AST
+from ..ast_node import ASTNode
 from .constants import NODE, BLOCK_REASON, ORIGIN_STATEMENT, NodeId
 from ._nodes_factory import NodesFactory
 from ._block_extractors import BlockInfo, extract_blocks_from_statement

diff --git a/aibolit/ast_framework/block_statement_graph/statement.py b/aibolit/ast_framework/block_statement_graph/statement.py
@@ -1,7 +1,7 @@
-from networkx import DiGraph
+from networkx import DiGraph  # type: ignore
 from typing import Callable, Iterator, Any, TYPE_CHECKING
 
-from veniq.ast_framework import ASTNode
+from ..ast_node import ASTNode
 from .constants import NODE, NodeId
 
 if TYPE_CHECKING:

diff --git a/test/ast_framework/test_block_statement_graph.py b/test/ast_framework/test_block_statement_graph.py
@@ -2,10 +2,10 @@
 from pathlib import Path
 from unittest import TestCase
 
-from veniq.ast_framework.block_statement_graph import build_block_statement_graph, Block, Statement
-from veniq.ast_framework.block_statement_graph.constants import BlockReason
-from veniq.ast_framework import AST, ASTNodeType
-from veniq.utils.ast_builder import build_ast
+from aibolit.ast_framework.block_statement_graph import build_block_statement_graph, Block, Statement
+from aibolit.ast_framework.block_statement_graph.constants import BlockReason
+from aibolit.ast_framework import AST, ASTNodeType
+from aibolit.utils.ast_builder import build_ast
 
 
 class BlockStatementTestCase(TestCase):