Henry65 commited on
Commit
21be054
·
1 Parent(s): bf6e4e8

Update RepoPipeline.py

Browse files
Files changed (1) hide show
  1. RepoPipeline.py +7 -6
RepoPipeline.py CHANGED
@@ -42,15 +42,16 @@ def extract_requirements(lines):
42
  """
43
  requirements_set = set()
44
  for line in lines:
 
45
  try:
46
- if line != "\n":
47
- if " == " in line:
48
- splitLine = line.split(" == ")
49
- else:
50
- splitLine = line.split("==")
51
- requirements_set.add(splitLine[0])
52
  except:
53
  pass
 
54
  return requirements_set
55
 
56
 
 
42
  """
43
  requirements_set = set()
44
  for line in lines:
45
+ line = line.replace("\n", "").strip()
46
  try:
47
+ if " == " in line:
48
+ splitLine = line.split(" == ")
49
+ elif "==" in line:
50
+ splitLine = line.split("==")
51
+ requirements_set.add(splitLine[0])
 
52
  except:
53
  pass
54
+
55
  return requirements_set
56
 
57