DavMelchi commited on
Commit
8b971f0
·
1 Parent(s): bd9f7f6

Add error handling for non-numeric cell codes in extract_code function

Browse files
apps/kpi_analysis/trafic_analysis.py CHANGED
@@ -41,7 +41,12 @@ class TraficAnalysis:
41
  ############### PROCESSING ###############
42
  def extract_code(name):
43
  name = name.replace(" ", "_") if isinstance(name, str) else None
44
- return int(name.split("_")[0]) if name and len(name) >= 10 else None
 
 
 
 
 
45
 
46
 
47
  def preprocess_2g(df: pd.DataFrame) -> pd.DataFrame:
 
41
  ############### PROCESSING ###############
42
  def extract_code(name):
43
  name = name.replace(" ", "_") if isinstance(name, str) else None
44
+ if name and len(name) >= 10:
45
+ try:
46
+ return int(name.split("_")[0])
47
+ except ValueError:
48
+ return None
49
+ return None
50
 
51
 
52
  def preprocess_2g(df: pd.DataFrame) -> pd.DataFrame: