From cb9e8170ac52c670feb5425c54befe19e3815560 Mon Sep 17 00:00:00 2001
From: bbfamily <bbfamily@126.com>
Date: Sat, 21 Oct 2017 00:18:08 +0800
Subject: [PATCH] fix similar code

---
 abupy/SimilarBu/ABuSimilar.py | 5 ++++-
 1 file changed, 4 insertions(+), 1 deletion(-)
 mode change 100755 => 100644 abupy/SimilarBu/ABuSimilar.py

diff --git a/abupy/SimilarBu/ABuSimilar.py b/abupy/SimilarBu/ABuSimilar.py
old mode 100755
new mode 100644
index fe2e05e..b25fcf3
--- a/abupy/SimilarBu/ABuSimilar.py
+++ b/abupy/SimilarBu/ABuSimilar.py
@@ -24,7 +24,7 @@
 from ..CoreBu.ABuEnv import EMarketDataSplitMode, EMarketTargetType
 from ..MarketBu import ABuSymbolPd
 from ..MarketBu.ABuMarket import split_k_market, all_symbol
-from ..MarketBu.ABuSymbol import IndexSymbol
+from ..MarketBu.ABuSymbol import IndexSymbol, Symbol
 from ..UtilBu.ABuDTUtil import consume_time
 from ..UtilBu.ABuProgress import do_clear_output
 from ..CoreBu.ABuEnvProcess import add_process_env_sig, AbuEnvProcess
@@ -102,6 +102,9 @@ def _find_similar(symbol, cmp_cnt=None, n_folds=2, start=None, end=None, show_cn
     :param show: 是否可视化最终top最相关的股票
     :param corr_type: ECoreCorrType对象，暂时支持皮尔逊，斯皮尔曼，＋－符号相关系数，移动时间加权相关系数
     """
+    if isinstance(symbol, Symbol):
+        # 如果传递的时Symbol对象，取value
+        symbol = symbol.value
     # 获取全市场symbol涨跌幅度pd.DataFrame对象
     market_change_df = _all_market_cg(symbol, cmp_cnt=cmp_cnt, n_folds=n_folds, start=start, end=end)
     if market_change_df is None: