          self.nodes = {}  # Key = key, val = Node
  
      def add_node(self, key):
          if key is None:
              raise Exception('Invalid key')
          if key in self.nodes:
              return self.nodes[key]
          self.nodes[key] = Node(key)
          return self.nodes[key]
  
 
</s> remove     "            raise Exception('Invalid key')\n",
 </s> add     "            raise TypeError('key cannot be None')\n", </s> remove         if neighbor is None:
            raise Exception('Invalid neighbor')
 </s> add         if neighbor is None or weight is None:
            raise TypeError('neighbor or weight cannot be None') </s> remove     def add_undirected_edge(self, source_key, dest_key, weight=0):
        if source_key is None or dest_key is None:
            raise Exception('Invalid key')
        self.add_edge(source_key, dest_key, weight)
        self.nodes[dest_key].add_neighbor(self.nodes[source_key],
                                          weight) </s> add  </s> remove         if neighbor is None or neighbor.key not in self.adj_nodes:
            raise Exception('Invalid neighbor')
 </s> add         if neighbor is None:
            raise TypeError('neighbor cannot be None')
        if neighbor.key not in self.adj_nodes:
            raise KeyError('neighbor not found') </s> remove     "        if neighbor is None:\n",
    "            raise Exception('Invalid neighbor')\n",
 </s> add     "        if neighbor is None or weight is None:\n",
    "            raise TypeError('neighbor or weight cannot be None')\n",
            raise TypeError('key cannot be None')
          self.nodes = {}  # Key = key, val = Node
  
      def add_node(self, key):
          if key is None:
            raise TypeError('key cannot be None')
          if key in self.nodes:
              return self.nodes[key]
          self.nodes[key] = Node(key)
          return self.nodes[key]
  
keep keep keep keep replace keep keep keep keep keep
 Update graph challenge (#117)

Use more specific exception types. Update constraints. Update add_undirected_edge to use add_edge. 
graphs_trees/graph/graph.py