From 789288c9b6695435311173ff77bdd639cf1f8f6b Mon Sep 17 00:00:00 2001 From: Andrew Yates Date: Fri, 7 Apr 2023 00:02:37 +0200 Subject: [PATCH] black update --- capreolus/reranker/POSITDRMM.py | 1 - docs/conf.py | 1 + pyproject.toml | 3 +++ setup.py | 1 + 4 files changed, 5 insertions(+), 1 deletion(-) create mode 100644 pyproject.toml diff --git a/capreolus/reranker/POSITDRMM.py b/capreolus/reranker/POSITDRMM.py index 6d70edcf9..bfbe34e78 100644 --- a/capreolus/reranker/POSITDRMM.py +++ b/capreolus/reranker/POSITDRMM.py @@ -39,7 +39,6 @@ def __init__(self, extractor, pipeline_config): self.hidden = self.init_hidden() def init_hidden(self): - return ( Variable(torch.zeros(4, self.batch_size, self.lstm_hidden_dim).to(device)), Variable(torch.zeros(4, self.batch_size, self.lstm_hidden_dim).to(device)), diff --git a/docs/conf.py b/docs/conf.py index 1e2707f0f..bcfbc4e95 100644 --- a/docs/conf.py +++ b/docs/conf.py @@ -18,6 +18,7 @@ sys.path.insert(0, os.path.abspath("../")) assert os.path.exists(os.path.abspath("../capreolus")) + # from https://packaging.python.org/guides/single-sourcing-package-version/ def read(rel_path): here = os.path.abspath(os.path.dirname(__file__)) diff --git a/pyproject.toml b/pyproject.toml new file mode 100644 index 000000000..b438d0541 --- /dev/null +++ b/pyproject.toml @@ -0,0 +1,3 @@ +[tool.black] +line-length = 130 +target_version = ['py37'] diff --git a/setup.py b/setup.py index 66b94d95a..ba007e4b6 100644 --- a/setup.py +++ b/setup.py @@ -21,6 +21,7 @@ def run(self): with open("README.md", "r") as fh: long_description = fh.read() + # from https://packaging.python.org/guides/single-sourcing-package-version/ def read(rel_path): here = os.path.abspath(os.path.dirname(__file__))