From b94239234a747f3c76683b4e187dbc87a26c7f72 Mon Sep 17 00:00:00 2001
From: shadiendrawis <40395873+shadiendrawis@users.noreply.github.com>
Date: Wed, 21 Nov 2018 16:09:04 +0200
Subject: [PATCH] Removed TF warning when training in a distributed setting
 (#133)

* removed TF warning when training in a distributed setting and changed package version

* revert version back to 0.11.0
---
 .gitignore                                                     | 1 +
 .../tensorflow_components/distributed_tf_utils.py              | 3 ++-
 2 files changed, 3 insertions(+), 1 deletion(-)

diff --git a/.gitignore b/.gitignore
index 2f46d8b..50bcf54 100644
--- a/.gitignore
+++ b/.gitignore
@@ -17,6 +17,7 @@ docs/site
 coach_env
 build
 rl_coach.egg*
+rl_coach_slim.egg*
 contrib
 test_log_*
 dist
diff --git a/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py b/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py
index 2ea216d..bbbbc0f 100644
--- a/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py
+++ b/rl_coach/architectures/tensorflow_components/distributed_tf_utils.py
@@ -95,7 +95,8 @@ def create_monitored_session(target: tf.train.Server, task_index: int,
         hooks=[],
         checkpoint_dir=checkpoint_dir,
         save_checkpoint_secs=checkpoint_save_secs,
-        config=config
+        config=config,
+        log_step_count_steps=0  # disable logging of steps to avoid TF warning during inference
     )
 
     return sess