update to 0.1.18

3 files changed, 41 insertions, 15 deletions

diff --git a/.SRCINFO b/.SRCINFO
index 47d788562e90..e94f5f947f50 100644
--- a/.SRCINFO
+++ b/.SRCINFO
@@ -1,12 +1,18 @@
 pkgbase = python-pgmpy
 	pkgdesc = Probabilistic Graphical Model library
-	pkgver = 0.1.17
+	pkgver = 0.1.18
 	pkgrel = 1
 	url = https://github.com/pgmpy/pgmpy
 	changelog = CHANGELOG.md
 	arch = any
 	license = MIT
+	checkdepends = python-pytest
+	checkdepends = python-mock
+	checkdepends = python-daft
 	makedepends = python-setuptools
+	makedepends = python-build
+	makedepends = python-installer
+	makedepends = python-wheel
 	makedepends = python-sphinx
 	makedepends = python-nbsphinx
 	makedepends = python-sphinx_rtd_theme
@@ -21,7 +27,7 @@ pkgbase = python-pgmpy
 	depends = python-statsmodels
 	depends = python-pytorch
 	depends = python-tqdm
-	source = python-pgmpy-0.1.17.tar.gz::https://github.com/pgmpy/pgmpy/archive/v0.1.17.tar.gz
-	sha256sums = 1b2bc18dfcdb0d1f7d966061978b3ac5771849da5861f2a80dbea5ed799e5b3b
+	source = python-pgmpy-0.1.18.tar.gz::https://github.com/pgmpy/pgmpy/archive/v0.1.18.tar.gz
+	sha256sums = 89b5d95e3f5f8e862a2a09c9ef0b7fbd62467803c550f1b8c19759d7ffc679d5
 
 pkgname = python-pgmpy
diff --git a/CHANGELOG.md b/CHANGELOG.md
index 86eea2529944..b264470f8a2e 100644
--- a/CHANGELOG.md
+++ b/CHANGELOG.md
@@ -4,6 +4,13 @@ All notable changes to this project will be documented in this file.
 The format is based on [Keep a Changelog](https://keepachangelog.com/en/1.0.0/),
 and this project adheres to [Semantic Versioning](https://semver.org/spec/v2.0.0.html).
 
+## [0.1.18] - 2021-03-30
+### Fixed
+1. Fixes `CausalInference.is_valid_backdoor_adjustment_set` to accept str arguments for `Z`.
+2. Fixes `BayesianNetwork.remove_cpd` to work with integral node names.
+3. Fixes `MPLP.map_query` to return the variable states instead of probability values.
+4. Fixes BIFWriter to generate output in standard BIF format.
+
 ## [0.1.17] - 2021-12-30
 ### Added
 1. Adds BayesianNetwork.states property to store states of all the variables.
diff --git a/PKGBUILD b/PKGBUILD
index 690c5b7637ef..938daa2a4ecb 100644
--- a/PKGBUILD
+++ b/PKGBUILD
@@ -1,7 +1,7 @@
 # Maintainer: Luis Martinez <luis dot martinez at disroot dot org>
 
 pkgname=python-pgmpy
-pkgver=0.1.17
+pkgver=0.1.18
 pkgrel=1
 pkgdesc="Probabilistic Graphical Model library"
 arch=('any')
@@ -21,29 +21,42 @@ depends=(
 	'python-tqdm')
 makedepends=(
 	'python-setuptools'
+	'python-build'
+	'python-installer'
+	'python-wheel'
 	'python-sphinx'
 	'python-nbsphinx'
 	'python-sphinx_rtd_theme')
-# checkdepends=('python-pytest' 'python-mock' 'python-daft')
+checkdepends=('python-pytest' 'python-mock' 'python-daft')
 changelog=CHANGELOG.md
 source=("$pkgname-$pkgver.tar.gz::$url/archive/v$pkgver.tar.gz")
-sha256sums=('1b2bc18dfcdb0d1f7d966061978b3ac5771849da5861f2a80dbea5ed799e5b3b')
+sha256sums=('89b5d95e3f5f8e862a2a09c9ef0b7fbd62467803c550f1b8c19759d7ffc679d5')
+
+prepare() {
+	cd "pgmpy-$pkgver"
+	sed -i '/"tests"/s/tests/*tests*/' setup.py
+}
 
 build() {
 	cd "pgmpy-$pkgver"
-	python setup.py build
-	cd docs
-	make man BUILDDIR=_build
+	python -m build --wheel --no-isolation
+	make -C docs man BUILDDIR=_build
 }
 
-# check() {
-# 	cd "pgmpy-$pkgver"
-# 	pytest
-# }
+check() {
+	cd "pgmpy-$pkgver"
+	pytest -x --disable-warnings
+}
 
 package() {
+	export PYTHONHASHSEED=0
 	cd "pgmpy-$pkgver"
-	PYTHONHASHSEED=0 python setup.py install --root="$pkgdir/" --optimize=1 --skip-build
-	install -Dm644 LICENSE -t "$pkgdir/usr/share/licenses/$pkgname/"
+	python -m installer --destdir="$pkgdir/" dist/*.whl
 	install -Dm644 docs/_build/man/pgmpy.1 -t "$pkgdir/usr/share/man/man1/"
+
+	local _site="$(python -c 'import site; print(site.getsitepackages()[0])')"
+	install -d "$pkgdir/usr/share/licenses/$pkgname/"
+	ln -s \
+		"$_site/pgmpy-$pkgver.dist-info/LICENSE" \
+		"$pkgdir/usr/share/licenses/$pkgname/"
 }