From 3041f491a4225ca592963b4be3c8685fc7ebf39f Mon Sep 17 00:00:00 2001
From: reterik <joeemailaddy@gmail.com>
Date: Wed, 13 Nov 2019 22:47:52 -0800
Subject: [PATCH] Update sample.py

Tensorflow fails with just sort namespace
---
 src/sample.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/src/sample.py b/src/sample.py
index c90ed28dc..da83bd4ce 100644
--- a/src/sample.py
+++ b/src/sample.py
@@ -25,7 +25,7 @@ def _top_k():
 def top_p_logits(logits, p):
     """Nucleus sampling"""
     batch, _ = logits.shape.as_list()
-    sorted_logits = tf.sort(logits, direction='DESCENDING', axis=-1)
+    sorted_logits = tf.contrib.framework.sort(logits, direction='DESCENDING', axis=-1)
     cumulative_probs = tf.cumsum(tf.nn.softmax(sorted_logits, axis=-1), axis=-1)
     indices = tf.stack([
         tf.range(0, batch),