Skip to content

Commit

Permalink
linting fixes
Browse files Browse the repository at this point in the history
  • Loading branch information
john681611 committed Sep 14, 2023
1 parent 6296b77 commit aa76908
Show file tree
Hide file tree
Showing 4 changed files with 2 additions and 5 deletions.
3 changes: 2 additions & 1 deletion application/cmd/cre_main.py
Original file line number Diff line number Diff line change
Expand Up @@ -533,9 +533,10 @@ def owasp_metadata_to_cre(meta_file: str):
"""
raise NotImplementedError("someone needs to work on this")


def populate_neo4j_db(cache: str):
logger.info(f"Populating neo4j DB: Connecting to SQL DB")
database = db_connect(path=cache)
logger.info(f"Populating neo4j DB: Populating")
database.neo_db.populate_DB(database.session)
logger.info(f"Populating neo4j DB: Complete")
logger.info(f"Populating neo4j DB: Complete")
1 change: 0 additions & 1 deletion application/database/db.py
Original file line number Diff line number Diff line change
Expand Up @@ -463,7 +463,6 @@ def load_cre_graph(cls, session) -> nx.Graph:

graph.add_edge(f"CRE: {lnk.cre}", f"Node: {str(lnk.node)}", ltype=lnk.type)
return graph



class Node_collection:
Expand Down
2 changes: 0 additions & 2 deletions application/tests/db_test.py
Original file line number Diff line number Diff line change
Expand Up @@ -1306,7 +1306,6 @@ def test_gap_analysis_duplicate_link_path_existing_higher(self, gap_mock):
}
self.assertEqual(collection.gap_analysis(["a", "b"]), expected)


def test_get_embeddings_by_doc_type_paginated(self):
"""Given: a range of embedding for Nodes and a range of embeddings for CREs
when called with doc_type CRE return the cre embeddings
Expand Down Expand Up @@ -1471,6 +1470,5 @@ def test_get_embeddings_by_doc_type(self):
self.assertEqual(tool_emb, {})



if __name__ == "__main__":
unittest.main()
1 change: 0 additions & 1 deletion cre.py
Original file line number Diff line number Diff line change
Expand Up @@ -193,7 +193,6 @@ def main() -> None:
help="populate the neo4j db",
)


args = parser.parse_args()

from application.cmd import cre_main
Expand Down

0 comments on commit aa76908

Please sign in to comment.