Skip to content

Commit

Permalink
code format
Browse files Browse the repository at this point in the history
  • Loading branch information
yangky11 committed Sep 15, 2024
1 parent 2f188be commit 1aa1b87
Show file tree
Hide file tree
Showing 7 changed files with 10 additions and 3 deletions.
1 change: 1 addition & 0 deletions check_data.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Sanity checks on the data.
"""

import os
from hashlib import md5

Expand Down
1 change: 1 addition & 0 deletions common.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Some utility functions.
"""

import re
import torch
import random
Expand Down
7 changes: 4 additions & 3 deletions preprocess_ruletaker.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Preprocess the RuleTaker dataset into a format similar to EntailmentBank.
"""

from common import *
from glob import glob
from lark import Lark
Expand Down Expand Up @@ -190,9 +191,9 @@ def main() -> None:
"context": context,
"proofs": proofs,
"answer": answer,
"depth": question["QDep"]
if answer != "Unknown"
else None,
"depth": (
question["QDep"] if answer != "Unknown" else None
),
}
)

Expand Down
1 change: 1 addition & 0 deletions prover/datamodule.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Dataloading for EntailmentBank and RuleTaker.
"""

from copy import deepcopy
from common import *
from prover.proof import Proof, InvalidProofStep
Expand Down
1 change: 1 addition & 0 deletions prover/evaluate.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Utilities for evaluation.
"""

from common import *
import argparse
import itertools
Expand Down
1 change: 1 addition & 0 deletions prover/proof.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Proof steps and proof trees.
"""

from common import *
import itertools

Expand Down
1 change: 1 addition & 0 deletions prover/search.py
Original file line number Diff line number Diff line change
@@ -1,6 +1,7 @@
"""
Proof graph for proof search.
"""

from common import *
from prover.proof import ProofStep
import networkx as nx
Expand Down

0 comments on commit 1aa1b87

Please # to comment.