From 60b7356d68b08b2d7b9a7ad4c0c27ebbe57a8f0a Mon Sep 17 00:00:00 2001
From: Anton Akhmerov <anton.akhmerov@gmail.com>
Date: Mon, 31 Dec 2018 01:43:11 +0100
Subject: [PATCH] internal function formatting cleanup
---
kwant/builder.py | 30 ++++++++++++++++--------------
1 file changed, 16 insertions(+), 14 deletions(-)
diff --git a/kwant/builder.py b/kwant/builder.py
index 044b951e..a74f6112 100644
--- a/kwant/builder.py
+++ b/kwant/builder.py
@@ -599,21 +599,23 @@ class BuilderLead(Lead):
return InfiniteSystem(self.builder, self.interface)
-# Check that a modes/selfenergy function has a keyword-only parameter
-# 'params', or takes '**kwargs'. If not, we wrap it
def _ensure_signature(func):
- parameters = inspect.signature(func).parameters
- has_params = bool(parameters.get('params'))
- has_kwargs = any(p.kind == inspect.Parameter.VAR_KEYWORD
- for p in parameters.values())
- if has_params or has_kwargs:
- return func
- else: # function conforming to old API: needs wrapping
-
- def wrapper(energy, args=(), *, params=None):
- return func(energy, args)
-
- return wrapper
+ """
+ Ensure that a modes/selfenergy function has a keyword-only parameter
+ ``params``, or takes ``**kwargs`` by potentially wrapping it.
+ """
+ parameters = inspect.signature(func).parameters
+ has_params = bool(parameters.get('params'))
+ has_kwargs = any(p.kind == inspect.Parameter.VAR_KEYWORD
+ for p in parameters.values())
+ if has_params or has_kwargs:
+ return func
+
+ # function conforming to old API: needs wrapping
+ def wrapper(energy, args=(), *, params=None):
+ return func(energy, args)
+
+ return wrapper
class SelfEnergyLead(Lead):
--
GitLab