From e507086d959a2d86002ce6787f5870d09ff815fd Mon Sep 17 00:00:00 2001
From: Yuning598 <1147936698@qq.com>
Date: Sat, 8 Aug 2026 12:31:31 +0800
Subject: [PATCH] Preserve size groups in ranked outputs

---
 chars_ciz/functions.py          | 1 +
 chars_ciz/impute_rank_output.py | 2 +-
 2 files changed, 2 insertions(+), 1 deletion(-)

diff --git a/chars_ciz/functions.py b/chars_ciz/functions.py
index b33b75a..9c33e83 100644
--- a/chars_ciz/functions.py
+++ b/chars_ciz/functions.py
@@ -442,6 +442,7 @@ def standardize(df: pl.DataFrame):
         'permno', 'date', 'datadate', 'gvkey', 'sic', 'count',
         'exchcd', 'shrcd', 'ffi49', 'ret', 'retadj', 'retx',
         'lag_me', 'ticker', 'conm', 'comnam', 'prc', 'shrout',
+        'size_grp',
         'primaryexch', 'conditionaltype', 'tradingstatusflg',
         'sharetype', 'securitytype', 'securitysubtype',
         'usincflg', 'issuertype',
diff --git a/chars_ciz/impute_rank_output.py b/chars_ciz/impute_rank_output.py
index 073f46c..63fa6c8 100644
--- a/chars_ciz/impute_rank_output.py
+++ b/chars_ciz/impute_rank_output.py
@@ -25,7 +25,7 @@
 OBS_VARS = [
     'gvkey', 'permno', 'jdate', 'ticker', 'conm', 'comnam',
     'sic', 'ret', 'retx', 'retadj',
-    'exchcd', 'shrcd', 'prc', 'shrout',
+    'exchcd', 'shrcd', 'prc', 'shrout', 'size_grp',
 ]
 
 ACCOUNTING_VARS = [