Skip to content

Commit 4633064

Browse files
committed
fix: replace strange list comprehension, remove ABC
1 parent a05a22e commit 4633064

1 file changed

Lines changed: 2 additions & 6 deletions

File tree

developer/gen-atomicq-dataset.py

Lines changed: 2 additions & 6 deletions
Original file line numberDiff line numberDiff line change
@@ -1,7 +1,5 @@
11
#!/usr/bin/env python3
22

3-
from abc import ABC
4-
53
import numpy as np
64
import pandas as pd
75

@@ -14,7 +12,7 @@ class OPQDataError(Exception):
1412
"""Our specific exception"""
1513

1614

17-
class OPQDataStorer(ABC):
15+
class OPQDataStorer():
1816
def __init__(self, s: System, lname: str) -> None:
1917
self._s = s
2018
self._lname = lname
@@ -34,7 +32,7 @@ def _cvt_op_df(self, opdict: dict, err_pos: int) -> pd.DataFrame:
3432
_c_dict_len = len(_c_dict)
3533
if _c_dict_len == 0:
3634
continue
37-
vearr = np.array([_c_dict.popitem()[1] for _ in range(_c_dict_len)])
35+
vearr = np.array(list(_c_dict.values()))
3836
if _c_dict_len == 1: # one H
3937
op_clean.loc[len(op_clean)] = [smid, 1, vearr[0, 0], vearr[0, 1]]
4038
elif _c_dict_len == 3: # three H. They are always symmetric.
@@ -60,8 +58,6 @@ def ass_id(self) -> str:
6058

6159
def _get_smi2uname(self) -> dict:
6260
mol = self._s.lipids[self._lname]
63-
smiles = mol.metadata["bioschema_properties"]["smiles"]
64-
print(smiles)
6561
s2u = {}
6662
for uname, aprops in mol.mapping_dict.items():
6763
if "SMILEIDX" in aprops:

0 commit comments

Comments
 (0)