fix arguments types

janalis · Nov 30, 2017 · 61221aa · 61221aa
1 parent 71ed41b
commit 61221aa
Show file tree

Hide file tree

Showing 2 changed files with 5 additions and 5 deletions.
diff --git a/src/Graphviz/Attributable.php b/src/Graphviz/Attributable.php
@@ -70,7 +70,7 @@ public function getAttributes(): array
      *
      * @param AttributeInterface[] $attributes
      */
-    public function setAttributes($attributes): void
+    public function setAttributes(array $attributes): void
     {
         foreach ($attributes as $attribute) {
             $this->attributes[$attribute->getId()] = $attribute;
@@ -101,7 +101,7 @@ public function getAttribute(string $id): ?AttributeInterface
      *
      * @return AttributeInterface
      */
-    public function createAttribute($id, $value): AttributeInterface
+    public function createAttribute(string $id, string $value): AttributeInterface
     {
         $attribute = new Attribute($id, $value);
         $this->attributes[$id] = $attribute;

diff --git a/src/Graphviz/Edgeable.php b/src/Graphviz/Edgeable.php
@@ -99,12 +99,12 @@ public function addEdgeFrom(ElementInterface $element, string $label = null): Ed
     /**
      * Remove edge from.
      *
-     * @param Vertex|Record $element
-     * @param null|string   $label
+     * @param ElementInterface $element
+     * @param null|string      $label
      *
      * @throws \RuntimeException
      */
-    public function removeEdgeFrom($element, string $label = null): void
+    public function removeEdgeFrom(ElementInterface $element, string $label = null): void
     {
         if (!$this->graph) {
             throw new \RuntimeException('Graph is not defined');