Skip to content

Namrata's Family, diffusion of working conditions in supply chains

Chesney's Namrata's Family, buyers who push good working conditions down their supply network.

  • Name: supply_chain_diffusion
  • Field: social/organizations
  • Formalism: agent-based system
  • References: Chesney 2021; Gold, Chesney, Gruchmann, and Trautrims 2020

Select this system

from skcomplex.systems import create_simulation, create_system, describe

system = create_system('supply_chain_diffusion')
entry = describe('supply_chain_diffusion')
trajectory = create_simulation('supply_chain_diffusion').prepare().run().trajectory_

Factory

@register_system(
    field='social/organizations',
    tags=('agents', 'network', 'supply chains', 'diffusion', 'labour standards'),
    references=('Chesney 2021', 'Gold, Chesney, Gruchmann, and Trautrims 2020'),
    defaults={'until': 10},
)
def supply_chain_diffusion(
    graph: object = None,
    *,
    buyers: int = 3,
    factories: int = 40,
    seeds: int = 2,
    random_state: int | np.random.Generator | None = 0,
) -> System:
    """Return Chesney's Namrata's Family, buyers who push good working conditions down their supply network.

    Goods flow from suppliers to buyers along the weighted edges of a directed network, `w_ji` from `j` to `i`. A buyer
    `i` that has adopted forces a supplier `j` to adopt when it is less beholden to the supplier than the supplier is
    to it, `V_ij = w_ji / w_->i < V_ji = w_ji / w_j->`, where `w_->i` is the total value that flows into `i` and
    `w_j->` the total value that flows out of `j`, with a strict inequality since factories resist by default
    (Chesney 2021 sections 4.9 and 4.10). A seed group of buyers adopts at the start, and in every time period the
    firms that adopted in the previous one push their suppliers. Since the condition reads `w_->i > w_j->`, a buyer
    with a single supplier never forces it, so diffusion needs buyers with two or more suppliers (section 4.11). The
    default network has `buyers` buyers and `factories` factories, every factory supplying one to three random firms
    listed before it with values from 1 to 10.

    Args:
        graph:
            A NetworkX directed graph of the supply network with the values of the goods as the edge attribute
            `weight`, or `None` for a random network.

        buyers:
            The number of buyers, the first nodes, of the default network.

        factories:
            The number of factories of the default network.

        seeds:
            The number of the first nodes that adopt at the start.

        random_state:
            The seed or generator of the default network.

    Returns:
        system:
            The system, with the group `firms` on the nodes of the network.

    Examples:
        >>> from skcomplex.systems.social.organizations import supply_chain_diffusion
        >>> supply_chain_diffusion(factories=10).elements[0].prepare().size_
        13
    """
    rng = np.random.default_rng(random_state)
    network = _supply_network(buyers, factories, rng) if graph is None else graph
    if not isinstance(network, nx.DiGraph):
        network = nx.DiGraph(network)
    nodes = list(network.nodes)
    adopted = np.zeros(len(nodes))
    adopted[:seeds] = 1.0
    firms = GroupElement(
        'firms',
        element=SupplierElement,
        size=len(nodes),
        values={
            'adopted': adopted,
            'inflow': [float(network.in_degree(node, weight='weight')) for node in nodes],
            'outflow': [float(network.out_degree(node, weight='weight')) for node in nodes],
        },
    )
    adoption = MapInteraction('adoption', target='firms', state='adopted', function=_adopt, form='agent')
    return System(elements=[firms], interactions=[adoption], space=NetworkSpace(network))

Components

Elements:

  • SupplierElement, from Element

Interactions:

  • MapInteraction
class SupplierElement(Element):
    """Firm of a supply chain, a buyer or a factory, with the value of the goods that flow into and out of it.

    Args:
        label:
            The label of the firm or the group.

        visual:
            How the agent is drawn, or `None` for `default_visual_`.

        adopted:
            Whether the firm has adopted good working conditions.

        inflow:
            The total value `w_->i` of the goods that flow into the firm from its suppliers.

        outflow:
            The total value `w_j->` of the goods that flow out of the firm to its buyers.

    Examples:
        >>> from skcomplex.systems.social.organizations import SupplierElement
        >>> SupplierElement('factory', outflow=6.0).outflow
        6.0
    """

    def __init__(
        self,
        label: str,
        visual: VisualSpec | None = VisualSpec('agents', {'color_by': 'adopted', 'color_range': (0.0, 1.0)}),
        *,
        adopted: Annotated[float, StateSpec(domain=FiniteEnumeratedSet([0, 1]))] = 0.0,
        inflow: Annotated[float, StateSpec(domain=RealSet.closed_open(0, oo))] = 0.0,
        outflow: Annotated[float, StateSpec(domain=RealSet.closed_open(0, oo))] = 0.0,
    ) -> None:
        super().__init__(label, visual=visual)
        self.adopted = adopted
        self.inflow = inflow
        self.outflow = outflow

    @classmethod
    def _get_view(cls) -> type[SupplierView]:
        """Return the class of the views of the agents, whose attributes are the states."""
        return SupplierView

class MapInteraction(Interaction):
    """Map of a state of discrete time, whose next value is a function, or the state plus it.

    In the mode `'replace'` the next value of the state is `f`, and in the mode `'increment'` it is the state plus `f`.
    Under the `'synchronous'` schedule every member of a bound group updates at once, from the states at the start of
    the step. Under the `'sequential'` schedule the members update one after the other, in their order, and under the
    `'random-sequential'` schedule in a new random order every step, each member reading the values the members before
    it set. Under the `'member'` schedule of the agent form, every member updates in its own activation, together with
    the rules of its element and the other member interactions, in the order the system lists them.

    In the `'symbolic'` form, `function(namespace)` returns a symbolic expression, or a vector of them for a vector
    state, where `namespace.target` holds the states and parameters of the target, `namespace.sources` the namespaces
    of the sources, and `namespace.t` the time, and may hold draws such as `normal('noise')`. In the `'agent'` form,
    the target is an agent group and `function(agent, system)` returns the value as a number, or as an array of the
    shape of a vector state, so it can read the agent, its neighbours, and the history and storage of the system view.

    Args:
        label:
            The label of the interaction, unique among the parts of its system.

        target:
            The path of the element or group whose state the map sets.

        state:
            The name of the state of the target.

        function:
            The function `f(namespace)` in the symbolic form, or `f(agent, system)` in the agent form.

        sources:
            The paths of the elements, or members of groups such as `cells[0]`, that the function reads.

        mode:
            `'replace'` or `'increment'`.

        schedule:
            `'synchronous'`, `'sequential'`, `'random-sequential'`, or `'member'` for the agent form.

        form:
            `'symbolic'` for update laws of the states of the target, or `'agent'` for a rule over an agent group.

        phase:
            The phase of the step the agent form runs in, or `None` for the default phase.

    Raises:
        ValidationError:
            The mode, the schedule, or the form is unknown, the symbolic form has the schedule `'member'`, the
            function is not callable, or the agent form returns a value that does not match the state, raised by
            `run` of the simulation.

    Examples:
        >>> from skcomplex.core.components.elements import Element
        >>> from skcomplex.core.components.interactions import MapInteraction
        >>> from skcomplex.core.components.spaces import DiscreteTime
        >>> from skcomplex.core.components.system import System
        >>> from skcomplex.core.engine.runtime import Simulation
        >>> from skcomplex.core.specs import StateSpec
        >>> from typing import Annotated, SupportsFloat
        >>> class ValueElement(Element):
        ...     def __init__(self, label, x: Annotated[float, StateSpec()] = 0.5):
        ...         super().__init__(label)
        ...         self.x = x
        >>> logistic = MapInteraction(
        ...     'map', target='value', state='x', function=lambda s: 3 * s.target.x * (1 - s.target.x)
        ... )
        >>> system = System(elements=[ValueElement('value')], interactions=[logistic], time=DiscreteTime())
        >>> Simulation(system).prepare().run(until=2).trajectory_.values('value.x').tolist()
        [0.5, 0.75, 0.5625]
        >>> agents = MapInteraction('map', target='walkers', state='x', function=lambda a, m: a.x + 1, form='agent')
        >>> type(agents.prepare().run(SimpleNamespace()).laws_[0]).__name__
        'RuleLaw'
        >>> from skcomplex.core.components.elements import GroupElement
        >>> class WeightsElement(Element):
        ...     def __init__(self, label, visual=None, *, w: Annotated[object, StateSpec(vector=True)] = 0.0):
        ...         super().__init__(label, visual=visual)
        ...         self.w = w
        >>> group = GroupElement('weights', element=WeightsElement, size=1, values={'w': [1.0, 2.0]})
        >>> doubling = MapInteraction(
        ...     'double', target='weights', state='w', function=lambda a, m: [2 * x for x in a.w], form='agent'
        ... )
        >>> trajectory = Simulation(System(elements=[group], interactions=[doubling])).prepare().run(until=2)
        >>> trajectory.trajectory_.agents('weights', 2).values['w'].tolist()
        [[4.0, 8.0]]
    """

    law_classes: ClassVar[tuple[type[BaseLaw], ...]] = (RuleLaw, UpdateLaw)

    def __init__(
        self,
        label: str,
        *,
        target: Annotated[str, RoleSpec(elements=(ElementKind.ANY,), states=('state',))],
        state: str,
        function: Callable[..., object],
        sources: Annotated[Sequence[str], RoleSpec(variadic=True, elements=(ElementKind.ANY,))] = (),
        mode: str = 'replace',
        schedule: str = 'synchronous',
        form: str = 'symbolic',
        phase: str | None = None,
    ) -> None:
        super().__init__(label)
        self.target = target
        self.state = state
        self.function = function
        self.sources = sources
        self.mode = mode
        self.schedule = schedule
        self.form = form
        self.phase = phase

    def _check(self) -> None:
        """Raise an error for an unknown mode, schedule, or form, or a function that cannot be called.

        Raises:
            ValidationError:
                The mode, the schedule, or the form is unknown, the symbolic form has the schedule `'member'`, or the
                function is not callable.
        """
        super()._check()
        self._check_variant('mode', self.mode, MODES, (BaseSpec,))
        for name, value, variants in (
            ('schedule', self.schedule, SCHEDULES),
            ('form', self.form, FORMS),
        ):
            if value not in variants:
                error_msg = f'The {name} of map {self.label!r} is {value!r}. Use one of {", ".join(variants)}.'
                raise ValidationError(error_msg)
        if self.schedule == 'member' and self.form != 'agent':
            error_msg = (
                f'The map {self.label!r} updates every member in its activation, which only the agent form does.'
            )
            raise ValidationError(error_msg)
        if not callable(self.function):
            error_msg = f'The function of map {self.label!r} must be callable, but it is {self.function!r}.'
            raise ValidationError(error_msg)

    def _apply_mode(self, current: Value, value: Value) -> Value:
        """Return the next value of a state from its current value and the value of the function, by the mode."""
        if self.mode == 'increment':
            return current + value
        if self.mode == 'replace':
            return value
        raise self._get_variant_error(self.mode, 'mode', '_apply_mode')

    def _get_vector(self, agent: AgentViewProtocol, system: SystemViewProtocol) -> Value:
        """Return the next value of a vector state of an agent."""
        current = np.asarray(getattr(agent, self.state), dtype=np.float64)
        value = np.asarray(self.function(agent, system), dtype=np.float64)
        if value.shape != current.shape:
            error_msg = (
                f'The function of map {self.label!r} returns the shape {value.shape} for the state {self.state!r} of '
                f'the shape {current.shape}. Return an array of the shape of the state.'
            )
            raise ValidationError(error_msg)
        return self._apply_mode(current, value)

    def _get_value(self, agent: AgentViewProtocol, system: SystemViewProtocol) -> Value:
        """Return the next value of the state of an agent, a number or the array of a vector state."""
        if np.ndim(getattr(agent, self.state)) > 0:
            return self._get_vector(agent, system)
        found = self.function(agent, system)
        if not isinstance(found, SupportsFloat):
            error_msg = f'The function of map {self.label!r} returns {found!r} for an agent. Return a number.'
            raise ValidationError(error_msg)
        value = float(found)
        return self._apply_mode(float(getattr(agent, self.state)), value)

    def _set_value(self, agent: AgentViewProtocol, value: Value) -> None:
        """Set the state of an agent to its next value."""
        setattr(agent, self.state, value)

    def _update_member(self, agent: AgentViewProtocol, system: SystemViewProtocol) -> None:
        """Set the state of one member to its next value."""
        if agent.alive:
            self._set_value(agent, self._get_value(agent, system))

    def update(self, system: SystemViewProtocol) -> None:
        """Set the state of every agent of the target to its next value, by the schedule.

        Args:
            system:
                The system view of the interaction.
        """
        self._update_agents(system, self.schedule, self._get_value, self._set_value)

    def _get_order(self) -> object:
        """Return the order of the members of the symbolic form: none, their own order, or a uniform draw."""
        if self.schedule == 'synchronous':
            return None
        if self.schedule == 'sequential':
            return SR(0)
        return uniform(re.sub(r'\W', '_', f'order_{self.label}'))

    def _get_laws(self, namespace: SimpleNamespace) -> list[BasePart]:
        """Return the update of the state to its next value, or the rule of the agent form.

        Args:
            namespace:
                The namespace of the symbols the laws can use, with `namespace.target` and `namespace.sources`.

        Returns:
            laws:
                The update law of the state, or the rule that runs the map of the agents.
        """
        if self.form == 'agent' and self.schedule == 'member':
            return [RuleLaw(self._update_member, phase=self.phase, members=self._get_role('target'))]
        if self.form == 'agent':
            return [RuleLaw(self.update, phase=self.phase)]
        value = getattr(self._get_member(namespace, 'target'), self.state)
        mapped = cast(Value, self.function(namespace))
        next_value = self._apply_mode(value, mapped)
        return [UpdateLaw(value, next_value, increment=self.mode == 'increment', order=self._get_order())]