From 28410b6206f4ec95ac66d1e4b392c83948a5fe7e Mon Sep 17 00:00:00 2001
From: Pavel Tomanek <99190809+pavelToman@users.noreply.github.com>
Date: Fri, 22 Nov 2024 14:51:22 +0100
Subject: [PATCH] Update easyconfigs.py - add exception for tiktoken version

---
 test/easyconfigs/easyconfigs.py | 2 ++
 1 file changed, 2 insertions(+)

diff --git a/test/easyconfigs/easyconfigs.py b/test/easyconfigs/easyconfigs.py
index 4b1c4dd770e..916196f9440 100644
--- a/test/easyconfigs/easyconfigs.py
+++ b/test/easyconfigs/easyconfigs.py
@@ -658,6 +658,8 @@ def check_dep_vars(self, gen, dep, dep_vars):
                 # tensorflow-probability version to TF version
                 ('2.8.4;', ['tensorflow-probability-0.16.0-']),
             ],
+            # vLLM has pinned dependency tiktoken == 0.6.0
+            'tiktoken': [('0.6.0;', ['vLLM-0.4.0-'])],
             # smooth-topk uses a newer version of torchvision
             'torchvision': [('0.11.3;', ['smooth-topk-1.0-20210817-'])],
             # for the sake of backwards compatibility, keep UCX-CUDA v1.11.0 which depends on UCX v1.11.0