fix comment/docs imports

Author: aboudreault

cassandra/datastax/graph/fluent/__init__.py

@@ -180,8 +180,8 @@ def traversal_source(session=None, graph_name=None, execution_profile=EXEC_PROFI
 
             .. code-block:: python
 
-                from dse.cluster import Cluster
-                from dse_graph import DseGraph
+                from cassandra.cluster import Cluster
+                from cassandra.datastax.graph.fluent import DseGraph
 
                 c = Cluster()
                 session = c.connect()
@@ -218,7 +218,7 @@ def create_execution_profile(graph_name):
         @staticmethod
         def batch(*args, **kwargs):
             """
-            Returns the :class:`dse_graph.query.TraversalBatch` object allowing to
+            Returns the :class:`cassandra.datastax.graph.fluent.query.TraversalBatch` object allowing to
             execute multiple traversals in the same transaction.
             """
             return _DefaultTraversalBatch(*args, **kwargs)

docs/graph_fluent.rst

@@ -233,7 +233,7 @@ which is not really verbose and quite annoying to repeat in a code base. Let's c
 
 Now, we can use our DSL that is a lot cleaner::
 
-  from dse_graph import DseGraph
+  from cassandra.datastax.graph.fluent import DseGraph
 
   # ...
   g = DseGraph.traversal_source(session=session, traversal_class=MyAppTraversalsource)