From 194a163bb5b569d0310b8ae3e98ca3d61c2d2351 Mon Sep 17 00:00:00 2001 From: Gao Enhao Date: Thu, 28 Dec 2023 01:34:27 +0800 Subject: [PATCH] [MNT] install prolog on win and fix a minor bug in hed --- .github/workflows/build-and-test.yaml | 9 ++++++--- examples/hed/bridge.py | 4 +++- 2 files changed, 9 insertions(+), 4 deletions(-) diff --git a/.github/workflows/build-and-test.yaml b/.github/workflows/build-and-test.yaml index 5b560ef..a5dfb01 100644 --- a/.github/workflows/build-and-test.yaml +++ b/.github/workflows/build-and-test.yaml @@ -21,9 +21,12 @@ jobs: python-version: ${{ matrix.python-version }} - name: Display python version run: python -c "import sys; print(sys.version)" - - name: Install prolog - run: | - sudo apt-get install swi-prolog + - name: Install SWI-Prolog on Ubuntu + if: matrix.os == 'ubuntu-latest' + run: sudo apt-get install swi-prolog + - name: Install SWI-Prolog on Windows + if: matrix.os == 'windows-latest' + run: choco install swi-prolog - name: Install package dependencies run: | python -m pip install --upgrade pip diff --git a/examples/hed/bridge.py b/examples/hed/bridge.py index f20f2eb..d81d829 100644 --- a/examples/hed/bridge.py +++ b/examples/hed/bridge.py @@ -233,7 +233,9 @@ class HedBridge(SimpleBridge): seems_good = self.check_rule_quality(rules, val_data, equation_len) if seems_good: - self.reasoner.kb.learned_rules.update({equation_len: rules}) + self.reasoner.kb.learned_rules.update( + {equation_len: rules, equation_len + 1: rules} + ) self.model.save( save_path=os.path.join(save_dir, f"eq_len_{equation_len}.pth") )