From 497bebd041cf1e789971d1c3187626dcbead494e Mon Sep 17 00:00:00 2001
From: Mike Clark <github@wassname.org>
Date: Sun, 18 Nov 2018 15:49:28 +0800
Subject: [PATCH] typo: remove hardcoding of K in _knn_bruteforce

I believe that you meant for this to be self.K not 4, please correct me if I am wrong.
---
 layers.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/layers.py b/layers.py
index fe9feaa..5b17d0d 100644
--- a/layers.py
+++ b/layers.py
@@ -81,7 +81,7 @@ class KnnBruteforce(Layer):
     if self.data_format == 'expanded':
       positions = _remove_dim(positions, 2)
 
-    NN, _, _ = _knn_bruteforce(positions, K=4)
+    NN, _, _ = _knn_bruteforce(positions, K=self.K)
     NN = tf.transpose(NN, [0, 2, 1])
 
     if self.data_format == 'expanded':