GraphDistanceCalculator._transform_mol() - Code Metrics - Inspection of "added to_smiles functions" - richlewis42/scikit-chem - Measure and Improve Code Quality continuously with Scrutinizer

Completed

Push — master ( 1baf98...c0c140 )

by Rich

created 2016-06-07 13:25 UTC

GraphDistanceCalculator._transform_mol() A

↳ Parent: GraphDistanceCalculator

Complexity

Conditions

Size

Total Lines

Duplication

Lines	0
Ratio	0 %

Importance

Changes	1
Bugs	1	Features	0

Metric	Value
cc	1
c	1
b	1
f	0
dl	0
loc	4
rs	10

#! /usr/bin/env python
#
# Copyright (C) 2015 Rich Lewis <[email protected]>
# License: 3-clause BSD

"""
## skchem.descriptors.atom

Module specifying atom based descriptor generators.
"""

import pandas as pd
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
import numpy as np
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3

from rdkit import Chem
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
from rdkit.Chem import Crippen
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
from rdkit.Chem import Lipinski
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
from rdkit.Chem import rdMolDescriptors, rdPartialCharges
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3
from rdkit.Chem.rdchem import HybridizationType
# .scrutinizer.yml
before_commands:
    - sudo pip install abc # Python2
    - sudo pip3 install abc # Python3

import functools

from .. import core
from ..data import PERIODIC_TABLE
from ..filters import ORGANIC

def element(a):


    """ Return the element """

    return a.GetSymbol()

def is_element(a, symbol='C'):


    """ Is the atom of a given element """
    return element(a) == symbol

element_features = {'is_{}'.format(e): functools.partial(is_element, symbol=e) for e in ORGANIC}


def is_h_acceptor(a):


    """ Is an H acceptor? """

    m = a.GetOwningMol()

    idx = a.GetIdx()
    return idx in [i[0] for i in Lipinski._HAcceptors(m)]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def is_h_donor(a):


    """ Is an H donor? """

    m = a.GetOwningMol()

    idx = a.GetIdx()
    return idx in [i[0] for i in Lipinski._HDonors(m)]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def is_hetero(a):


    """ Is a heteroatom? """

    m = a.GetOwningMol()

    idx = a.GetIdx()
    return idx in [i[0] for i in Lipinski._Heteroatoms(m)]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def atomic_number(a):


    """ Atomic number of atom """

    return a.GetAtomicNum()

def atomic_mass(a):


    """ Atomic mass of atom """

    return a.mass

def explicit_valence(a):


    """ Explicit valence of atom """
    return a.GetExplicitValence()

def implicit_valence(a):


    """ Implicit valence of atom """

    return a.GetImplicitValence()

def valence(a):


    """ returns the valence of the atom """

    return explicit_valence(a) + implicit_valence(a)

def formal_charge(a):


    """ Formal charge of atom """

    return a.GetFormalCharge()

def is_aromatic(a):


    """ Boolean if atom is aromatic"""

    return a.GetIsAromatic()

def num_implicit_hydrogens(a):


    """ Number of implicit hydrogens """

    return a.GetNumImplicitHs()

def num_explicit_hydrogens(a):


    """ Number of explicit hydrodgens """

    return a.GetNumExplicitHs()

def num_hydrogens(a):


    """ Number of hydrogens """

    return num_implicit_hydrogens(a) + num_explicit_hydrogens(a)

def is_in_ring(a):


    """ Whether the atom is in a ring """

    return a.IsInRing()

def crippen_log_p_contrib(a):


    """ Hacky way of getting logP contribution. """

    idx = a.GetIdx()
    m = a.GetOwningMol()

    return Crippen._GetAtomContribs(m)[idx][0]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def crippen_molar_refractivity_contrib(a):


    """ Hacky way of getting molar refractivity contribution. """

    idx = a.GetIdx()
    m = a.GetOwningMol()

    return Crippen._GetAtomContribs(m)[idx][1]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def tpsa_contrib(a):


    """ Hacky way of getting total polar surface area contribution. """

    idx = a.GetIdx()
    m = a.GetOwningMol()

    return rdMolDescriptors._CalcTPSAContribs(m)[idx]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def labute_asa_contrib(a):


    """ Hacky way of getting accessible surface area contribution. """

    idx = a.GetIdx()
    m = a.GetOwningMol()

    return rdMolDescriptors._CalcLabuteASAContribs(m)[0][idx]
class MyParent:
    def __init__(self):
        self._x = 1;
        self.y = 2;

class MyChild(MyParent):
    def some_method(self):
        return self._x    # Ok, since accessed from a child class

class AnotherClass:
    def some_method(self, instance_of_my_child):
        return instance_of_my_child._x   # Would be flagged as AnotherClass is not
                                         # a child class of MyParent

def gasteiger_charge(a, force_calc=False):


    """ Hacky way of getting gasteiger charge """

    res = a.props.get('_GasteigerCharge', None)
    if res and not force_calc:
        return float(res)
    else:
        idx = a.GetIdx()

        m = a.GetOwningMol()

        rdPartialCharges.ComputeGasteigerCharges(m)
        return float(a.props['_GasteigerCharge'])

def electronegativity(a):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    return PERIODIC_TABLE.loc[a.atomic_number, 'pauling_electronegativity']

def first_ionization(a):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    return PERIODIC_TABLE.loc[a.atomic_number, 'first_ionisation_energy']

def group(a):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    return PERIODIC_TABLE.loc[a.atomic_number, 'group']

def period(a):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    return PERIODIC_TABLE.loc[a.atomic_number, 'period']

def is_hybridized(a, hybrid_type=HybridizationType.SP3):


    """ Hybridized as type hybrid_type, default SP3 """

    return str(a.GetHybridization()) is hybrid_type

hybridization_features = {'is_' + n + '_hybridized': functools.partial(is_hybridized, hybrid_type=n) for n in HybridizationType.names}


atom_features = {

    'atomic_number': atomic_number,
    'atomic_mass': atomic_mass,
    'formal_charge': formal_charge,
    'gasteiger_charge': gasteiger_charge,
    'electronegativity': electronegativity,
    'first_ionisation': first_ionization,
    'group': group,
    'period': period,
    'valence': valence,
    'is_aromatic': is_aromatic,
    'num_hydrogens': num_hydrogens,
    'is_in_ring': is_in_ring,
    'log_p_contrib': crippen_log_p_contrib,
    'molar_refractivity_contrib': crippen_molar_refractivity_contrib,
    'is_h_acceptor': is_h_acceptor,
    'is_h_donor': is_h_donor,
    'is_heteroatom': is_hetero,
    'total_polar_surface_area_contrib': tpsa_contrib,
    'total_labute_accessible_surface_area': labute_asa_contrib,
}
atom_features.update(element_features)
atom_features.update(hybridization_features)

class AtomFeatureCalculator(object):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    def __init__(self, features='all', max_atoms=50):
        if features == 'all':
            features = atom_features
        self.features = pd.Series(features)
        self.max_atoms = max_atoms

    @property
    def feature_names(self):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        return self.features.index

    def transform(self, obj):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if isinstance(obj, core.Atom):
            return self._transform_atom(obj)
        elif isinstance(obj, core.Mol):
            return self._transform_mol(obj)
        elif isinstance(obj, pd.Series):
            return self._transform_series(obj)
        elif isinstance(obj, pd.DataFrame):
            return self._transform_series(obj.structure)
        elif isinstance(obj, (tuple, list)):
            return self._transform_series(obj)
        else:
            raise NotImplementedError

    def _transform_atom(self, atom):
        return self.features.apply(lambda f: f(atom))

    def _transform_mol(self, mol):
        return pd.DataFrame(self.transform(a) for a in mol.atoms)

    def _transform_series(self, data):
        X = np.repeat(np.nan,

                      len(self.feature_names) * self.max_atoms * len(data))\
                      .reshape((len(data), self.max_atoms, len(self.feature_names)))
        for i, mol in enumerate(data):
            X[i, :len(mol.atoms), :] = self._transform_mol(mol)
        res = pd.Panel(X, items=data.index)
        res.minor_axis = self.features.index
        res.major_axis.name = 'atom_idx'
        return res


    def __call__(self, *args, **kwargs):
        return self.transform(*args, **kwargs)

class GraphDistanceCalculator(object):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""

    def __init__(self, max_atoms=50):
        self.max_atoms = max_atoms

    def transform(self, obj):
class SomeClass:
    def some_method(self):
        """Do x and return foo."""
        if isinstance(obj, core.Atom):
            return self._transform_atom(obj)

        elif isinstance(obj, core.Mol):
            return self._transform_mol(obj)
        elif isinstance(obj, pd.Series):
            return self._transform_series(obj)
        elif isinstance(obj, pd.DataFrame):
            return self._transform_series(obj.structure)
        elif isinstance(obj, (tuple, list)):
            return self._transform_series(obj)
        else:
            raise NotImplementedError

    def _transform_mol(self, mol):
        res = np.repeat(np.nan, self.max_atoms ** 2).reshape(self.max_atoms, self.max_atoms)
        res[:len(mol.atoms), :len(mol.atoms)] = Chem.GetDistanceMatrix(mol)
        return res

    def _transform_series(self, ser):
        res = pd.Panel(np.array([self.transform(mol) for mol in ser]), items=ser.index)
        res.major_axis.name = res.minor_axis.name = 'atom_idx'

    def __call__(self, *args, **kwargs):
        return self.transform(*args, **kwargs)


Push — master ( 1baf98...c0c140 )

GraphDistanceCalculator._transform_mol() A

Complexity

Size

Duplication

Importance

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1. Missing Dependencies

2. Missing init.py files

1			#! /usr/bin/env python
2			#
3			# Copyright (C) 2015 Rich Lewis <[email protected]>
4			# License: 3-clause BSD
5
6			"""
7			## skchem.descriptors.atom
8
9			Module specifying atom based descriptor generators.
10			"""
11
12			import pandas as pd
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `pandas` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
13			import numpy as np
			0 ignored issues – show Configuration introduced 2016-06-03 13:34 UTC by Report Bug Copy Issue Report The import `numpy` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
14
15			from rdkit import Chem
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `rdkit` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
16			from rdkit.Chem import Crippen
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `rdkit.Chem` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
17			from rdkit.Chem import Lipinski
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `rdkit.Chem` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
18			from rdkit.Chem import rdMolDescriptors, rdPartialCharges
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `rdkit.Chem` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
19			from rdkit.Chem.rdchem import HybridizationType
			0 ignored issues – show Configuration introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The import `rdkit.Chem.rdchem` could not be resolved. This can be caused by one of the following: 1. Missing Dependencies This error could indicate a configuration issue of Pylint. Make sure that your libraries are available by adding the necessary commands. # .scrutinizer.yml before_commands: - sudo pip install abc # Python2 - sudo pip3 install abc # Python3 Tip: We are currently not using virtualenv to run pylint, when installing your modules make sure to use the command for the correct version. 2. Missing __init__.py files This error could also result from missing `__init__.py` files in your module folders. Make sure that you place one file in each sub-folder. Loading history...
20
21			import functools
22
23			from .. import core
24			from ..data import PERIODIC_TABLE
25			from ..filters import ORGANIC
26
27			def element(a):
			0 ignored issues – show Coding Style Naming introduced 2016-01-19 23:07 UTC by Report Bug Copy Issue Report The name `a` does not conform to the argument naming conventions (`[a-z_][a-z0-9_]{2,30}$`). This check looks for invalid names for a range of different identifiers. You can set regular expressions to which the identifiers must conform if the defaults do not match your requirements. If your project includes a Pylint configuration file, the settings contained in that file take precedence. To find out more about Pylint, please refer to their site. Loading history...
28
29			""" Return the element """
30
31			return a.GetSymbol()
32

richlewis42 / scikit-chem

Push — master ( 1baf98...c0c140 )

GraphDistanceCalculator._transform_mol() A

Complexity

Size

Duplication

Importance

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

1. Missing Dependencies

2. Missing __init__.py files

Duplication Side-by-Side

Filter issues like

2. Missing init.py files

2. Missing init.py files

2. Missing init.py files

2. Missing init.py files

2. Missing init.py files

2. Missing init.py files

2. Missing init.py files