Improve comments, show output of unit tests

Author: ludovicc

python-distributed-kmeans/Dockerfile

@@ -10,7 +10,7 @@ COPY distributed_kmeans.py /src/distributed_kmeans.py
 COPY tests/unit /src/tests/
 
 WORKDIR /src
-RUN pytest tests/
+RUN pytest tests/ -x --ff --capture=no
 
 
 FROM hbpmip/python-mip-sklearn:latest

python-distributed-kmeans/distributed_kmeans.py

@@ -66,7 +66,7 @@
 
 @utils.catch_user_error
 def intermediate_kmeans():
-    """Calculate kNN locally."""
+    """Calculate k-Means locally."""
     # Read inputs
     logging.info("Fetching data...")
     inputs = io_helper.fetch_data()
@@ -104,7 +104,7 @@ def intermediate_kmeans():
     num_iter = 0
     not_converged = True
 
-    # Run kNN locally
+    # Run k-Means locally
     # Have each site compute k initial clusters locally
     local_centroids = local.initialize_own_centroids(X, k)