From 5004fe54c0f505bf1a194b95f880c4fbf42329b0 Mon Sep 17 00:00:00 2001
From: Eitan Turok <eitan.turok@gmail.com>
Date: Wed, 25 Sep 2024 23:58:48 +0000
Subject: [PATCH] better comment

---
 tests/models/utils/test_tp_strategy.py | 3 ++-
 1 file changed, 2 insertions(+), 1 deletion(-)

diff --git a/tests/models/utils/test_tp_strategy.py b/tests/models/utils/test_tp_strategy.py
index f7d7b48a12..d1e48babcf 100644
--- a/tests/models/utils/test_tp_strategy.py
+++ b/tests/models/utils/test_tp_strategy.py
@@ -13,7 +13,8 @@
 
 
 def test_ffn_tp_strategy_layer_plan():
-    # Actual layer plan
+
+    # Actual layer plan from tp_strategy=fnn
     tp_config = {
         'strategy': 'ffn',
     }